在 CentOS 上安装 PyTorch 并不算难,但相比 Ubuntu,确实更容易遇到环境依赖问题,尤其是 GPU 版本。下面我按“难度从低到高”给你讲清楚,并给出最稳妥的安装方案。
✅ CPU 版 PyTorch:很简单,10 分钟搞定
⚠️ GPU 版 PyTorch:中等难度,主要卡在 NVIDIA 驱动和 CUDA
❌ 老版本 CentOS(6 / 旧 7):不建议折腾
✅ 推荐环境
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
按提示安装后:
source ~/.bashrc
conda create -n torch python=3.9
conda activate torch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
✅ 这是最简单、最不容易出错的方式
python -c "import torch; print(torch.__version__)"
✅ NVIDIA 显卡
✅ 已安装:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
python -c "import torch; print(torch.cuda.is_available())"
如果返回 True,说明成功 ✅
现象
version `GLIBC_2.17' not found
✅ 解决
CentOS 默认 Python 2.7 / 3.6
✅ 解决
会直接报错
✅ 解决
pip install torch torchvision torchaudio
⚠️ 需要:
| 场景 | 难度 | 建议 |
|---|---|---|
| CPU + conda | ⭐ | ✅ 强烈推荐 |
| GPU + conda | ⭐⭐ | ✅ 推荐 |
| pip + 系统 Python | ⭐⭐⭐⭐ | ❌ 不推荐 |
| CentOS 6 | ⭐⭐⭐⭐⭐ | ❌ 放弃 |
你可以直接告诉我:
1️⃣ CentOS 版本(cat /etc/centos-release)
2️⃣ 是否有 GPU(nvidia-smi)
3️⃣ Python 版本要求
4️⃣ 是否用 conda / pip / Docker
我可以一步一步给你写命令,包括驱动和 CUDA 安装。