温馨提示×

centos如何重装pytorch

小樊
46
2026-10-06 08:47:33
栏目: 智能运维

在 CentOS 上“重装 PyTorch”通常分为几种情况,下面按最常见、最安全的方式给你说明。


一、先确认当前 PyTorch 安装方式

先看看你原来是怎么装的:

pip show torch

或(如果你用 conda):

conda list torch

二、使用 pip 重装 PyTorch(推荐)

1️⃣ 卸载旧版本

pip uninstall torch torchvision torchaudio -y

2️⃣ 安装新版本(CPU 示例)

pip install torch torchvision torchaudio

3️⃣ 安装 CUDA 版本(如 CentOS 有 NVIDIA 显卡)

先确认 CUDA 版本:

nvidia-smi

例如 CUDA 11.8:

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

三、使用 conda 重装(推荐服务器环境)

1️⃣ 卸载

conda uninstall pytorch torchvision torchaudio

2️⃣ 安装(CPU)

conda install pytorch torchvision torchaudio cpuonly -c pytorch

3️⃣ CUDA 版本(如 CUDA 11.8)

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

四、CentOS 常见问题

✅ 1. glibc 版本过低

CentOS 7 可能报错:

GLIBC_2.17 not found

解决:

  • 使用 conda(自带依赖)
  • 或升级系统到 CentOS 8 / Rocky / AlmaLinux

✅ 2. 缺少依赖

sudo yum install -y python3-devel gcc

五、验证安装

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

如果有 GPU:

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

如果你愿意,可以告诉我:

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

我可以给你精准命令。

0 踩