ubuntu pytorch如何升版本

在 Ubuntu 上升级 PyTorch 版本，通常是通过 pip 或 conda 来完成的。下面分情况说明。

一、先确认当前环境

1. 查看 PyTorch 版本

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

2. 确认安装方式

pip show torch
# 或
conda list torch

二、使用 pip 升级 PyTorch（最常见）

1. 升级到最新稳定版

pip install --upgrade torch torchvision torchaudio

2. 指定版本（例如 2.1.0）

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0

3. 使用 CUDA 版本（示例：CUDA 11.8）

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

官方安装命令可在 https://pytorch.org/get-started/locally/ 生成

三、使用 conda 升级 PyTorch

1. 升级到最新版

conda update pytorch torchvision torchaudio -c pytorch

2. 指定 CUDA 版本

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

四、虚拟环境（推荐）

避免系统污染，建议使用虚拟环境：

venv 示例

python -m venv pytorch_env
source pytorch_env/bin/activate
pip install --upgrade torch

conda 示例

conda create -n torch2 python=3.10
conda activate torch2
conda install pytorch -c pytorch

五、常见问题

1. 版本不兼容

• Python 版本过低（如 3.7 不支持新 PyTorch）
• CUDA 驱动版本过低

查看 CUDA 驱动：

nvidia-smi

2. 升级后报错

可先卸载再安装：

pip uninstall torch torchvision torchaudio
pip install torch torchvision torchaudio

如果你告诉我：

• Ubuntu 版本
• Python 版本
• 是否用 GPU（CUDA）
• 当前 PyTorch 版本

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