可以,CentOS 完全支持安装 PyTorch。
不过需要注意 CentOS 版本、Python 版本 以及 是否需要 GPU(CUDA),这会影响具体安装方式。下面给你一个清晰、实用的说明。
⚠️ CentOS 8 官方已 EOL,建议迁移到 Rocky Linux / AlmaLinux,但 PyTorch 仍可正常安装。
cat /etc/centos-release
CentOS 自带 Python 可能较旧,建议使用 pyenv 或 conda。
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.10
conda activate torch
pip install torch torchvision torchaudio
nvidia-smi
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
或 pip:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
python -c "import torch; print(torch.__version__)"
GPU 验证:
python -c "import torch; print(torch.cuda.is_available())"
PyTorch 新版本可能要求:
ldd --version
✅ 解决方案:
例如:
pip install torch==1.13.1 torchvision==0.14.1
sudo yum install -y gcc gcc-c++ make
| 方式 | 说明 |
|---|---|
| yum 直接装 PyTorch | ❌ 不可用 |
| 系统 Python | ❌ 容易冲突 |
| 源码编译 | ❌ 没必要,除非定制 |
✅ CentOS 可以安装 PyTorch
✅ 最佳方案:Miniconda + pip / conda
✅ CentOS 7 建议用稍旧版 PyTorch
✅ GPU 需要提前配置 NVIDIA 驱动和 CUDA
如果你愿意,可以告诉我:
我可以给你 一条完全适配你环境的安装命令 ✅