在 CentOS 上安装 PyTorch,通常可以按下面 5 步 完成(以 CPU 版为例,GPU 版会多一步装驱动):
sudo yum update -y
CentOS 7 一般自带 Python 2,建议装 Python 3:
sudo yum install -y python3 python3-pip
验证:
python3 --version
pip3 --version
pip3 install virtualenv
virtualenv pytorch_env
source pytorch_env/bin/activate
pip3 install torch torchvision torchaudio
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
python3
>>> import torch
>>> print(torch.__version__)
>>> print(torch.cuda.is_available()) # GPU 时为 True
最少 5 步:
如果你告诉我 CentOS 版本(7 / 8 / Stream) 和 是否用 GPU,我可以给你更精确的一条命令安装方案。