可以,CentOS 完全可以安装 PyTorch,但需要注意 系统版本、Python 版本、CUDA 驱动 等环境是否匹配。下面给你一个清晰、实用的说明。
PyTorch 官方支持:
| PyTorch 版本 | 推荐 Python |
|---|---|
| ≥ 2.0 | Python 3.8 – 3.11 |
| 1.x | Python 3.7 – 3.9 |
✅ 建议使用 Python 3.8 / 3.9
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 -y
conda activate torch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
⚠️ CUDA 版本需与 NVIDIA 驱动匹配
python3 -m pip install torch torchvision torchaudio
或指定 CUDA:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
python
import torch
print(torch.__version__)
print(torch.cuda.is_available())
✅ 输出类似:
2.1.0
True
CentOS 7 默认 gcc 4.8,可能导致编译失败
✅ 解决:
sudo yum install centos-release-scl
sudo yum install devtoolset-9
scl enable devtoolset-9 bash
glibc 2.17如报错,可:
✅ 用 conda / pip --user ✅ 不需要 root 也能完整使用 PyTorch
✅ CentOS 完全可以安装 PyTorch ✅ 推荐用 Conda ✅ CPU / GPU 都支持 ✅ 注意 Python 和 CUDA 版本匹配
如果你愿意,可以告诉我:
cat /etc/centos-release)我可以给你 一条完全适配你服务器的安装命令 ✅