在 CentOS 上安装 PyTorch,一般有两种常见方式:pip 安装(最常用、最简单)和 conda 安装。下面我按推荐程度给你说明。
CentOS 7/8 默认有 Python,但版本可能较低,建议先确认:
python3 --version
pip3 --version
如果没有:
sudo yum install python3 python3-pip -y
pip3 install --upgrade pip
pip3 install torch torchvision torchaudio
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
其他 CUDA 版本可去官网查看:
https://pytorch.org/get-started/locally/
python3 -c "import torch; print(torch.__version__)"
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
安装完成后重启终端或:
source ~/.bashrc
conda create -n pytorch python=3.10
conda activate pytorch
CPU:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
GPU(CUDA 11.8):
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
PyTorch 新版可能要求 glibc ≥ 2.17(CentOS 7 满足),但 CentOS 6 不支持。
sudo yum install -y libgomp
pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
✅ 新手 / 服务器:pip 安装
✅ 多项目 / 环境隔离:conda 安装
✅ 有 NVIDIA 显卡:选 CUDA 版本
如果你能告诉我:
我可以给你精确的命令。