linux怎么升级pytorch

linux

小樊

2026-10-04 11:40:42

栏目: 智能运维

在 Linux 上升级 PyTorch，常见有 pip 和 conda 两种方式，下面分别说明。

一、使用 pip 升级 PyTorch

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

pip install --upgrade torch torchvision torchaudio

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

到 PyTorch 官网获取对应命令，例如：

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

conda activate your_env_name

conda update pytorch torchvision torchaudio -c pytorch

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

pip install --user --upgrade torch

或使用虚拟环境（推荐）。

pip uninstall torch torchvision torchaudio

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

如果你能告诉我：

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