Ubuntu中如何更新PyTorch库

Ubuntu中更新PyTorch库的常用方法

一、使用pip更新PyTorch

pip是Python默认的包管理工具，适用于大多数Ubuntu环境（包括虚拟环境）。

1. 打开终端：通过快捷键Ctrl + Alt + T启动终端。
2. 检查当前PyTorch版本：运行pip show torch，查看已安装的版本信息。
3. 升级pip本身（可选但推荐）：运行pip install --upgrade pip，确保pip为最新版本，避免兼容性问题。
4. 更新PyTorch及相关库：
   • 升级到最新版本：运行pip install --upgrade torch torchvision torchaudio，这会自动下载并安装PyTorch、TorchVision、TorchAudio的最新稳定版。
   • 升级到指定版本：若需要特定版本（如1.12.0），可运行pip install torch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0（替换为所需版本号）。
5. 验证更新：再次运行pip show torch，确认Version字段显示为新版本；或在Python中执行import torch; print(torch.__version__)，打印版本号。

二、使用conda更新PyTorch（适用于Anaconda/Miniconda用户）

conda是Anaconda的包管理工具，更适合管理复杂的依赖关系（尤其是GPU环境）。

1. 打开终端：同样使用Ctrl + Alt + T。
2. 激活虚拟环境（若有）：运行conda activate your_env_name（将your_env_name替换为你的环境名称），确保在目标环境中操作。
3. 更新conda本身（可选）：运行conda update conda，保持conda版本最新。
4. 更新PyTorch及相关库：
   • GPU版本：若使用CUDA（如11.8），运行conda update pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch（将11.8替换为你的CUDA版本，可通过nvcc --version查看）。
   • CPU版本：若无需GPU加速，运行conda install pytorch torchvision torchaudio cpuonly -c pytorch。
5. 验证更新：运行conda list torch，检查torch行的版本号是否为新版本；或在Python中执行import torch; print(torch.__version__)。

三、注意事项

• 备份项目：更新前建议备份代码和数据，防止依赖冲突导致项目无法运行。
• 虚拟环境：务必在目标虚拟环境中操作（如venv或conda环境），避免影响系统全局Python环境。
• CUDA兼容性：若使用GPU版本，需确保CUDA Toolkit版本与PyTorch版本匹配（如PyTorch 2.0需CUDA 11.8及以上），可参考PyTorch官网的兼容性表。
• 卸载旧版本（可选）：若遇到依赖问题，可先卸载旧版本再重新安装（pip：pip uninstall torch torchvision torchaudio；conda：conda remove pytorch torchvision torchaudio）。