Ubuntu PyTorch如何更新版本

Ubuntu 更新 PyTorch 的简明步骤

一 准备与检查

• 确认当前环境：先激活你的虚拟环境（如使用 conda），再查看已装版本与 CUDA 信息。
  • 查看版本：python -c “import torch; print(torch.version)”
  • 查看 CUDA 运行时：python -c “import torch; print(torch.version.cuda)”
  • 查看是否可用 GPU：python -c “import torch; print(‘CUDA 可用:’, torch.cuda.is_available())”
• 建议做法：在独立的虚拟环境（venv 或 conda）中操作，避免影响系统或其他项目依赖。

二 使用 pip 更新

• 常规升级（保持现有 CUDA 版本不变）：pip install --upgrade torch torchvision torchaudio
• 指定渠道升级（如需要特定 CUDA 版本，示例为 CUDA 11.7）：pip install --upgrade torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
• 仅 CPU 环境：pip install --upgrade torch torchvision torchaudio
• 升级后验证：再次运行上面的版本与 CUDA 检查命令，确认版本号与 CUDA 可用性。

三 使用 Conda 更新

• 激活环境：conda activate <your_env>
• 常规升级：conda update pytorch torchvision torchaudio -c pytorch
• 指定渠道升级（示例为 CUDA 11.7）：conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
• 升级后验证：python -c “import torch; print(‘CUDA 可用:’, torch.cuda.is_available())”

四 常见问题与处理

• 升级失败或版本不一致：先卸载后重装（注意同时处理 torch、torchvision、torchaudio）。
  • pip 卸载：pip uninstall torch torchvision torchaudio
  • conda 卸载：conda remove pytorch torchvision torchaudio
  • 然后按上面的安装命令重新安装对应版本。
• 需要回退或安装指定版本：
  • pip：pip install torch==<版本号> torchvision==<版本号> torchaudio==<版本号>
  • conda：conda install pytorch=<版本号> torchvision=<版本号> torchaudio=<版本号> -c pytorch
  • 版本号可在 PyTorch 官网获取，建议与项目依赖矩阵匹配。
• 镜像与网络问题：国内用户可临时使用镜像源加速（如清华/中科大），或在网络稳定时切回官方源；更新完成后用 pip freeze 或 conda list 核对已装版本。