CentOS 上运行 PyTorch 是适合的,但有一些需要注意的地方,下面分情况说清楚:
✅ 非常适合以下情况
PyTorch 官方也提供 Linux x86_64 的 wheel 和 conda 包,CentOS 完全兼容。
如果你要用 CUDA:
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
✅ Conda(强烈推荐)
conda create -n torch python=3.10
conda activate torch
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
✅ 验证:
import torch
print(torch.cuda.is_available())
❌ 以下情况不太合适:
CentOS 适合“跑 PyTorch”,不适合“折腾 PyTorch”
如果你愿意,可以告诉我:
我可以给你一套最稳妥的安装方案。