温馨提示×

PyTorch在Ubuntu上的依赖如何解决

小樊
60
2025-08-11 22:21:37
栏目: 智能运维

在Ubuntu上安装PyTorch依赖可参考以下方法:

使用pip安装

  • 先更新系统包列表:sudo apt update
  • 安装Python和pip:sudo apt install python3 python3-pip
  • 根据CUDA版本选择安装命令,如无GPU用pip3 install torch torchvision torchaudio;有CUDA则用对应版本命令,如CUDA 11.7用pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

使用conda安装

  • 安装Miniconda或Anaconda。
  • 创建新环境:conda create -n pytorch_env python=3.8
  • 激活环境:conda activate pytorch_env
  • 安装PyTorch,无GPU用conda install pytorch torchvision torchaudio cpuonly -c pytorch;有CUDA则用conda install pytorch torchvision torchaudio cudatoolkit=对应版本 -c pytorch

安装完成后,可通过import torchprint(torch.__version__)以及print(torch.cuda.is_available())验证安装是否成功。

0