在 CentOS 上,PyTorch 是兼容 CUDA 的,但需要注意几个关键点:
PyTorch 并不是“自动适配所有 CUDA”,而是要选对版本:
| PyTorch 版本 | 支持 CUDA |
|---|---|
| PyTorch 1.12 | CUDA 10.2 / 11.3 / 11.6 |
| PyTorch 1.13 | CUDA 11.6 / 11.7 |
| PyTorch 2.0 / 2.1 | CUDA 11.7 / 11.8 |
| PyTorch 2.2+ | CUDA 11.8 / 12.1 |
安装时需明确选择,例如:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
CentOS 上常缺这些:
gcc / g++(CUDA 需要合适版本)glibc 版本(CUDA 12 要求 ≥ 2.17)libstdc++检查:
nvidia-smi
nvcc --version
gcc --version
import torch
print(torch.cuda.is_available())
print(torch.version.cuda)
如果你愿意,可以告诉我:
我可以直接给你一套可复制的安装命令。