PyTorch在Ubuntu上如何进行版本管理

在Ubuntu上管理PyTorch版本，可通过以下方式操作：

一、安装/更新PyTorch

• pip方式

  1. 卸载旧版本：pip uninstall torch torchvision torchaudio
  2. 安装指定版本（需搭配CUDA版本）：
     • GPU版本：pip install torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
     • CPU版本：pip install torch torchvision torchaudio

• conda方式

  1. 卸载旧版本：conda remove pytorch torchvision torchaudio
  2. 安装指定版本：
     • GPU版本：conda install pytorch==2.0.1 torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
     • CPU版本：conda install pytorch==2.0.1 torchvision torchaudio cpuonly -c pytorch

二、版本切换与验证

• 查看当前版本：

  import torch
  print(torch.__version__)  # 查看PyTorch版本
  print(torch.version.cuda)  # 查看CUDA版本（若安装GPU版）
  

• 切换虚拟环境：

  • 激活环境：source venv_name/bin/activate（venv）或 conda activate env_name（conda）
  • 重新安装对应环境的PyTorch版本即可完成切换。

三、注意事项

• CUDA兼容性：需确保PyTorch版本与CUDA版本匹配（如PyTorch 2.0.1需CUDA 11.8）。
• 环境隔离：建议使用虚拟环境（venv/conda）避免版本冲突。
• 驱动检查：通过nvidia-smi确认GPU驱动支持安装的CUDA版本。

参考来源：