debian

PyTorch在Debian上的安装步骤是什么

小樊
40
2025-07-20 11:04:23
栏目: 智能运维

在Debian系统上安装PyTorch,可以按照以下步骤进行:

方法一:使用pip安装

  1. 更新系统包列表
sudo apt update
  1. 安装Python和pip(如果还没有安装):
sudo apt install python3 python3-pip
  1. 安装PyTorch
    • CPU版本
      pip3 install torch torchvision torchaudio
      
    • 支持CUDA的版本(例如CUDA 11.7):
      pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
      

方法二:使用conda安装

  1. 安装Miniconda(如果还没有安装):

    wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
    bash Miniconda3-latest-Linux-x86_64.sh
    
  2. 创建并激活conda环境(可选):

    conda create -n pytorch_env python=3.9
    conda activate pytorch_env
    
  3. 安装PyTorch

    • CPU版本
      conda install pytorch torchvision torchaudio cpuonly -c pytorch
      
    • 支持CUDA的版本(例如CUDA 11.7):
      conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
      

验证安装

安装完成后,可以通过以下命令验证PyTorch是否安装成功:

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果安装了CUDA版本,应该返回True

0
看了该问题的人还看了