在 CentOS 上运行 PyTorch 出错很常见,原因通常包括系统版本过旧、缺少依赖库、CUDA 驱动不匹配、Python 环境混乱等。下面我按「排查步骤 + 常见错误 + 解决方法」给你一个实用指南。
请先执行以下命令,把结果发我也可以帮你精确判断:
cat /etc/centos-release
python --version
pip --version
nvidia-smi
nvcc --version
python -c "import torch; print(torch.__version__)"
典型错误:
ImportError: libgomp.so.1: cannot open shared object file
ImportError: libcudart.so.11.0: cannot open shared object file
✅ 解决:
sudo yum install -y libgomp
sudo yum install -y mesa-libGL
如果是 CUDA 相关:
sudo yum install -y cuda-cudart-11-7
CentOS 7 默认 glibc 太旧(2.17),新版 PyTorch 可能跑不了。
✅ 解决方案(推荐):
conda create -n torch python=3.9
conda activate torch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
常见原因:
✅ 检查:
nvidia-smi
python -c "import torch; print(torch.cuda.is_available())"
✅ 重装对应 CUDA 的 PyTorch(示例):
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
多发生在 老 CPU + 新 PyTorch
✅ 解决:
pip install torch==1.13.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
通常是:
✅ 强烈建议用虚拟环境:
python -m venv venv
source venv/bin/activate
pip install torch
✅ 用 Miniconda(最省心)
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
conda create -n pt python=3.9
conda activate pt
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
为了更精准帮你,可以贴出:
我可以直接给你可复制的解决方案。