centos下pytorch如何配置

在 CentOS 下配置 PyTorch，通常分为 CPU 版 和 GPU（CUDA）版 两种场景。下面给你一套通用且稳妥的配置步骤。

一、前置准备（必须）

1. 更新系统

sudo yum update -y

2. 安装基础依赖

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

二、安装 Python（推荐 3.8 / 3.9 / 3.10）

CentOS 自带 Python 可能是 2.x 或旧 3.x，不建议直接用系统 Python。

方式一（推荐）：使用 Miniconda

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

安装完成后：

source ~/.bashrc
conda create -n torch python=3.9 -y
conda activate torch

三、安装 PyTorch

✅ CPU 版本（最通用）

pip install torch torchvision torchaudio

✅ GPU 版本（需要 NVIDIA 显卡 + 驱动）

1. 确认 CUDA 驱动已安装

nvidia-smi

能看到显卡信息说明驱动 OK。

2. 安装对应 CUDA 的 PyTorch（示例 CUDA 11.8）

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

常见版本对应关系：

• CUDA 11.7 → cu117
• CUDA 11.8 → cu118
• CUDA 12.1 → cu121

四、验证是否安装成功

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

五、常见问题（CentOS 特有）

1. GLIBC 版本过低

CentOS 7 默认 glibc 较老，某些新版 PyTorch 可能不支持。
✅ 解决方案：

• 使用 Conda 安装 PyTorch（推荐）

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

2. 找不到 libgomp

sudo yum install -y libgomp

3. 多用户服务器

• 使用 conda 或 virtualenv
• 不要直接 pip install 到系统 Python

六、推荐组合（最稳）

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

• CentOS 版本（7 / 8 / Stream）
• 是否用 GPU
• Python 版本需求

我可以给你精确到命令的配置方案。