在Debian系统上安装PyTorch,可以按照以下步骤进行:
sudo apt update
sudo apt install python3 python3-pip
pip3 install torch torchvision torchaudio
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
安装Miniconda(如果还没有安装):
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
创建并激活conda环境(可选):
conda create -n pytorch_env python=3.9
conda activate pytorch_env
安装PyTorch:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
安装完成后,可以通过以下命令验证PyTorch是否安装成功:
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 如果安装了CUDA版本,应该返回True