Ubuntu上PyTorch版本如何更新 - 问答

Ubuntu系统更新PyTorch版本的常用方法

准备工作
打开终端（快捷键Ctrl+Alt+T），首先升级pip至最新版本以避免兼容性问题：
```
pip install --upgrade pip
```
若使用虚拟环境，请提前通过source your_env/bin/activate激活目标环境。
卸载旧版本（可选但推荐）
为避免旧版本残留导致的依赖冲突，建议先卸载现有PyTorch及相关组件：
```
pip uninstall torch torchvision torchaudio
```
安装最新版本PyTorch
- GPU版本（需匹配CUDA版本）：
  访问PyTorch官网获取当前支持的CUDA版本（如CUDA 11.8），然后运行对应命令（以CUDA 11.8为例）：
```
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
```
- CPU版本（无GPU支持）：
  直接运行以下命令安装CPU-only版本：
```
pip install torch torchvision torchaudio
```
验证更新结果
运行Python交互环境，输入以下代码检查版本：
```
import torch
print(torch.__version__)  # 输出当前PyTorch版本号
```

准备工作
打开终端，激活目标conda环境（若有）：
```
conda activate your_env_name
```
建议先更新conda至最新版本：
```
conda update conda
```
卸载旧版本（可选但推荐）
清理现有PyTorch及关联组件，减少冲突风险：
```
conda remove pytorch torchvision torchaudio cudatoolkit
```
安装最新版本PyTorch
- GPU版本（需匹配CUDA版本）：
  通过conda指定CUDA工具包版本（如CUDA 11.7）安装：
```
conda update pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
```
- CPU版本（无GPU支持）：
  使用cpuonly标签安装：
```
conda update pytorch torchvision torchaudio cpuonly -c pytorch
```
验证更新结果
在终端运行以下命令，查看已安装的PyTorch版本：
```
conda list torch  # 确认torch包的版本信息
```

0 赞

0 踩