centos上pytorch怎么配置

在 CentOS 上配置 PyTorch，通常分为 CPU 版 和 GPU（CUDA）版 两种情况。下面按常见、稳妥的方式一步步说明（以 CentOS 7 / 8 为例）。

一、前期准备

1. 更新系统

sudo yum update -y

2. 安装基础依赖

sudo yum install -y gcc gcc-c++ make wget bzip2

二、安装 Python（推荐用 Miniconda）

CentOS 自带 Python 版本较老，强烈建议用 Conda。

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. 创建环境

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

三、安装 PyTorch

✅ CPU 版（最简单）

pip install torch torchvision torchaudio

验证：

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

✅ GPU 版（CUDA，需 NVIDIA 显卡）

1. 查看显卡驱动

nvidia-smi

查看右上角的 CUDA Version（如 11.8、12.1）

2. 安装对应 PyTorch

例如 CUDA 11.8：

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

CUDA 12.1：

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

验证：

import torch
print(torch.cuda.is_available())  # True

四、常见问题

1. GLIBC 版本过低（CentOS 7 常见）

报错：version 'GLIBC_2.17' not found

解决：

• 升级系统（不推荐）
• 使用 Conda 安装 PyTorch（推荐）

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

2. 缺少 libgomp

sudo yum install -y libgomp

3. pip 很慢

pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

五、推荐组合（省心）

如果你告诉我：

• CentOS 版本
• 是否有 GPU
• Python 版本需求

我可以给你 精确的一条命令安装方案。