在 CentOS 上安装 PyTorch 常见坑不少,主要集中在 系统版本、Python 环境、CUDA 驱动、依赖库 这几块。下面按常见踩坑点给你总结一下:
✅ 建议:
conda 或 pyenv 管理 Pythonconda create -n torch python=3.10
conda activate torch
✅ 建议:
PyTorch 对 CUDA 版本要求很严:
| PyTorch | CUDA |
|---|---|
| 2.0+ | 11.7 / 11.8 |
| 2.1+ | 11.8 / 12.1 |
❌ 常见错误:
✅ 检查:
nvidia-smi
nvcc --version
✅ 推荐直接用官方命令装:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
cuXXX 版本✅ CPU 版本:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
❌ 报错示例:
version `GLIBC_2.28' not found
✅ 解决:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
CentOS 7 默认 libstdc++.so.6 太旧
✅ 检查:
strings /usr/lib64/libstdc++.so.6 | grep GLIBCXX
✅ 解决:
devtoolsetyum install centos-release-scl
yum install devtoolset-11
scl enable devtoolset-11 bash
pip install --upgrade pip
pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
✅ CentOS 强烈推荐 conda
❌ pip 容易踩系统库坑
import torch
print(torch.__version__)
print(torch.cuda.is_available())
✅ 用 conda
✅ Python ≥ 3.9
✅ CUDA 版本对得上
✅ 别硬刚系统 glibc
✅ 没 GPU 就装 CPU 版
如果你愿意,可以告诉我:
我可以直接给你 一条可复制的安装命令。