Ubuntu系统中PyTorch更新与维护指南

在Ubuntu系统中更新和维护PyTorch的指南如下：

更新PyTorch

使用pip更新

1. 卸载旧版本：

pip uninstall torch torchvision torchaudio

2. 检查当前版本：

pip show torch

3. 升级pip（如果需要）：

pip install --upgrade pip

4. 安装最新版本：

pip install --upgrade torch torchvision torchaudio

或者指定版本：

pip install torch==<version> torchvision==<version> torchaudio==<version>

5. 验证更新：

pip show torch

使用conda更新

1. 激活conda环境（如果有的话）：

conda activate your_environment_name

2. 卸载旧版本：

conda remove pytorch torchvision torchaudio cudatoolkit

3. 安装最新版本：

conda update pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

注意：cudatoolkit的版本应与你安装的CUDA版本相匹配。 4. 验证更新：

conda list torch

使用系统包管理器（不推荐）

1. 卸载旧版本：

sudo apt-get remove python3-torch python3-torchvision python3-torchaudio

2. 添加PyTorch PPA：

sudo add-apt-repository ppa:deadsnakes/ppa
sudo apt-get update

3. 安装最新版本：

sudo apt-get install python3-torch python3-torchvision python3-torchaudio

维护PyTorch

创建和激活虚拟环境

• 使用conda：

conda create -n pytorch_env python=3.9
conda activate pytorch_env

• 使用venv：

python3 -m venv pytorch_env
source pytorch_env/bin/activate

验证安装

• 导入PyTorch并检查CUDA支持：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

注意事项

• 在更新PyTorch之前，建议备份你的项目代码和数据。
• 确保你的CUDA和cuDNN版本与新的PyTorch版本兼容。
• 如果你在虚拟环境中工作，确保在更新PyTorch之前激活相应的虚拟环境。

通过以上步骤，你应该能够在Ubuntu系统上成功更新和维护PyTorch。