在 CentOS 上安装 PyTorch,推荐通过 Conda 或 pip 安装,下面给你几种常见且稳妥的方式(含 CPU / GPU)。
cat /etc/centos-release
建议 CentOS 7 或 8,CentOS 6 已不推荐。
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 -y
conda activate torch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
✅ 需要:
python3 -m pip install --upgrade pip
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
python
import torch
print(torch.__version__)
print(torch.cuda.is_available())
✅ 输出示例:
2.1.0
True # 或 False(CPU)
GLIBC_2.17 not found检查:
nvidia-smi
若没有输出,说明:
使用国内镜像:
pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
| 环境 | 推荐 |
|---|---|
| 新手 | Conda + CPU |
| 服务器 | Conda + CUDA |
| 生产 | pip + 虚拟环境 |
如果你愿意,可以告诉我:
我可以给你 一条精确的安装命令 ✅