温馨提示×

centos装pytorch要几步

小樊
46
2026-10-12 05:41:17
栏目: 智能运维

在 CentOS 上安装 PyTorch,通常可以按下面 5 步 完成(以 CPU 版为例,GPU 版会多一步装驱动):


1️⃣ 更新系统

sudo yum update -y

2️⃣ 安装 Python 和 pip

CentOS 7 一般自带 Python 2,建议装 Python 3:

sudo yum install -y python3 python3-pip

验证:

python3 --version
pip3 --version

3️⃣(推荐)创建虚拟环境

pip3 install virtualenv
virtualenv pytorch_env
source pytorch_env/bin/activate

4️⃣ 安装 PyTorch

  • CPU 版
pip3 install torch torchvision torchaudio
  • GPU 版(CUDA)
    先确认 CUDA 已安装,然后去官网选命令,例如:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

5️⃣ 验证安装

python3
>>> import torch
>>> print(torch.__version__)
>>> print(torch.cuda.is_available())  # GPU 时为 True

✅ 总结

最少 5 步:

  1. 更新系统
  2. 装 Python/pip
  3. 建虚拟环境(可选)
  4. 装 PyTorch
  5. 验证

如果你告诉我 CentOS 版本(7 / 8 / Stream) 和 是否用 GPU,我可以给你更精确的一条命令安装方案。

0 踩