在 CentOS 上选择 PyTorch 版本,主要受 系统版本、Python 版本、CUDA 驱动、CPU/GPU 影响。下面按常见情况给你一个兼容性说明(截至 2024 年,PyTorch 2.x 为主流)。
torch 1.13 / 2.0torch 1.13 + CUDA 11.7/11.8python3.8+torch 2.0 / 2.1 / 2.2torch 2.2 / 2.3 + CUDA 12.x| PyTorch 版本 | Python 支持 |
|---|---|
| 1.10–1.13 | 3.7 – 3.10 |
| 2.0 | 3.8 – 3.11 |
| 2.1 / 2.2 | 3.8 – 3.11 |
| 2.3+ | 3.8 – 3.12 |
⚠️ CentOS 7 默认无 Python 3.8+,需用:
condaSoftware Collections (scl)| CentOS | CUDA | PyTorch |
|---|---|---|
| 7 | 11.6 / 11.7 / 11.8 | ≤ 2.1 |
| 8 | 11.8 / 12.x | 2.0+ |
| Rocky 9 | 12.x | 2.2+ |
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
Python 3.9
torch==1.13.1+cu117
CUDA 11.7
Python 3.10
torch==2.1.0+cu118
CUDA 11.8
Python 3.11
torch==2.3.0+cu121
Q:CentOS 7 能装 PyTorch 2.2 吗?
A:不推荐,官方 wheel 已不支持,需源码编译,容易失败。
Q:没有 GPU 能跑吗?
A:可以,CPU 版完全可用,只是慢。
Q:conda 还是 pip?
A:CentOS 强烈推荐 conda,避免系统库冲突。
如果你能告诉我:
我可以给你精确到命令的安装方案。