CentOS怎么配置PyTorch

在 CentOS 上配置 PyTorch，一般分三步：准备环境 → 安装 PyTorch → 验证是否可用（尤其是 GPU）。下面我按 最常见、最稳妥 的方式给你讲，适用于 CentOS 7 / 8（Stream 也类似）。

一、准备系统环境

1️⃣ 更新系统（可选但推荐）

sudo yum update -y

2️⃣ 安装基础依赖

sudo yum install -y gcc gcc-c++ make \
    openssl-devel bzip2-devel libffi-devel \
    zlib-devel wget curl

二、安装 Python（强烈建议用 Miniconda）

⚠️ CentOS 自带 Python 太老，不建议直接用

1️⃣ 下载 Miniconda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

2️⃣ 安装

bash Miniconda3-latest-Linux-x86_64.sh

一路 yes 即可，安装完成后执行：

source ~/.bashrc

3️⃣ 创建 PyTorch 专用环境（推荐）

conda create -n torch python=3.10 -y
conda activate torch

三、安装 PyTorch（CPU / GPU）

✅ 方案一：CPU 版本（最简单）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

✅ 方案二：GPU 版本（有 NVIDIA 显卡）

1️⃣ 确认你有 NVIDIA 显卡

lspci | grep -i nvidia

2️⃣ 安装 NVIDIA 驱动（如未安装）

sudo yum install -y epel-release
sudo yum install -y nvidia-driver

验证：

nvidia-smi

⚠️ 如果没有输出，说明驱动没装成功，需要先处理驱动。

3️⃣ 安装 CUDA（推荐 11.8 或 12.1）

例如 CUDA 11.8：

conda install nvidia/label/cuda-11.8.0::cuda-toolkit

4️⃣ 安装 GPU 版 PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

四、验证 PyTorch 是否安装成功

进入 Python：

python

✅ CPU 验证

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出：

False

✅ GPU 验证

import torch
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))

输出类似：

True
NVIDIA GeForce RTX 3090

五、常见问题（CentOS 特有）

❌ 1. glibc 版本太低

CentOS 7 默认 glibc 2.17，部分新 PyTorch 可能不支持
✅ 解决：

• 使用 conda 安装（推荐）
• 或升级系统到 CentOS Stream 8/9

❌ 2. conda 很慢

可换国内源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --set show_channel_urls yes

❌ 3. 服务器无 GUI

✅ 完全没问题，PyTorch 在服务器上运行非常常见

六、推荐组合（稳定）

如果你愿意，可以告诉我：

• ✅ CentOS 版本
• ✅ 是否有 GPU
• ✅ 是本地还是服务器
• ✅ 是否用 Docker

我可以给你 完全定制的一套安装命令 ✅