怎样在centos用pytorch

在 CentOS 上使用 PyTorch，一般流程如下（以 CentOS 7/8 为例）：

一、准备工作

1. 更新系统

sudo yum update -y

2. 安装 Python 和 pip

CentOS 默认可能是 Python 2，建议用 Python 3：

sudo yum install -y python3 python3-pip

检查：

python3 --version
pip3 --version

二、创建虚拟环境（推荐）

pip3 install virtualenv
mkdir pytorch_env && cd pytorch_env
virtualenv venv
source venv/bin/activate

三、安装 PyTorch

方式一：CPU 版本（无 GPU）

pip install torch torchvision torchaudio

方式二：GPU 版本（CUDA）

先确认你的 CUDA 版本：

nvidia-smi

例如 CUDA 11.8：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

官方安装命令生成器：
https://pytorch.org/get-started/locally/

四、验证是否安装成功

python

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

五、常见问题（CentOS 特有）

1. 缺少依赖库

如果报错（如 libgomp、GLIBC）：

sudo yum install -y libgomp glibc

2. CentOS 7 太老

CentOS 7 默认 gcc 太低，可能不支持新 PyTorch：

sudo yum install -y centos-release-scl
sudo yum install -y devtoolset-9
scl enable devtoolset-9 bash

3. 网络问题

可换国内源：

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

六、简单示例

import torch

x = torch.rand(3, 3)
print(x)

如果你告诉我：

• CentOS 版本（7 还是 8/9）
• 是否有 GPU
• 是否用于训练还是推理

我可以给你更精确的安装方案。