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

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

方法一：使用pip安装

1. 更新系统包列表：

sudo apt update

2. 安装Python和pip（如果还没有安装）：

sudo apt install python3 python3-pip

3. 安装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