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

ubuntu

小樊

2025-09-13 01:34:24

栏目: 智能运维

在Ubuntu上管理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版本即可完成切换。

参考来源：

最新问答