是的,CentOS 完全支持 PyTorch,但需要注意 Python 版本 和 系统依赖。下面分情况说明:
这是最省心、兼容性最好的方式。
# 安装 Miniconda 或 Anaconda
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
# 创建 Python 环境(建议 3.9 或 3.10)
conda create -n torch python=3.10
conda activate torch
# 安装 PyTorch(CPU 示例)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
# GPU 示例(CUDA 11.8)
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
✅ 优点
PyTorch 官方要求:
| PyTorch 版本 | Python 要求 |
|---|---|
| 2.x | ≥ 3.8 |
| 1.13 | ≥ 3.7 |
CentOS 7 默认 不满足,需要升级 Python。
python3 -m pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
✅ 解决方式:使用 conda
检查:
nvidia-smi
| 环境 | 建议 |
|---|---|
| CentOS 7 + 无 GPU | ✅ conda + CPU |
| CentOS 7 + GPU | ⚠️ 驱动 + conda |
| CentOS Stream 8/9 | ✅ 官方支持 |
| 服务器生产环境 | ✅ conda / Docker |
如果你愿意,可以告诉我:
我可以给你 一条最适合你的安装命令。