ubuntu pytorch如何升级版本

在 Ubuntu 上升级 PyTorch 版本，通常取决于你是用 pip 还是 conda 安装的。下面分别说明。

一、使用 pip 升级 PyTorch

1. 查看当前版本

python -c "import torch; print(torch.__version__)"

2. 升级到最新稳定版

pip install --upgrade torch torchvision torchaudio

3. 指定 CUDA 版本（推荐，GPU 用户）

例如 CUDA 11.8：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

其他常见版本：

• CUDA 12.1：cu121
• CPU 版：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

二、使用 conda 升级 PyTorch

1. 查看当前版本

conda list torch

2. 升级到最新版

conda update pytorch torchvision torchaudio -c pytorch

3. 指定 CUDA 版本

例如 CUDA 11.8：

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

三、虚拟环境（强烈推荐）

如果你还没用虚拟环境，建议新建一个：

python -m venv venv
source venv/bin/activate
pip install torch torchvision torchaudio

或用 conda：

conda create -n torch_env python=3.10
conda activate torch_env

四、验证升级是否成功

python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

五、常见注意事项

• 先卸载旧版本（如有冲突）：

pip uninstall torch torchvision torchaudio

• 确保 NVIDIA 驱动 支持你要装的 CUDA 版本
• 不要用 apt 安装 PyTorch（版本很旧）

如果你告诉我：

• Ubuntu 版本
• 是否用 GPU
• 当前 PyTorch 版本
• 用 pip 还是 conda

我可以给你精确的一条命令。