温馨提示×

centos pytorch怎么升级

小樊
46
2026-09-06 12:33:48
栏目: 智能运维

CentOS 上升级 PyTorch,常见做法取决于你最初是用 pip 还是 conda 安装的。下面分别说明。


一、先确认当前环境

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

如果是 conda 环境:

conda info --envs
conda activate 你的环境名

二、使用 pip 升级(最常见)

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. 使用国内镜像(CentOS 常见)

pip install --upgrade torch torchvision torchaudio \
  -i https://pypi.tuna.tsinghua.edu.cn/simple

4. 如果有 CUDA(非常重要)

查看 CUDA 版本:

nvidia-smi
nvcc --version

例如 CUDA 11.8:

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

三、使用 conda 升级

conda update pytorch torchvision torchaudio -c pytorch

指定 CUDA:

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

四、常见问题(CentOS 特有)

1. GLIBC 版本过低

CentOS 7 可能报错:

ImportError: /lib64/libc.so.6: version `GLIBC_2.17` not found

解决方式:

  • 使用 conda 安装 PyTorch(推荐)
  • 或升级系统(不推荐生产环境)

2. 权限问题

不要用 root 直接装,推荐:

pip install --user torch --upgrade

或放在虚拟环境中。


五、验证升级是否成功

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

如果你能告诉我:

  • CentOS 版本(7 / 8 / Stream)
  • Python 版本
  • 是否使用 GPU
  • 安装方式(pip / conda)

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

0