centos pytorch需要啥环境

在 CentOS 上跑 PyTorch，需要准备的环境主要分为 系统依赖、Python 环境、CUDA（可选） 三部分。下面按常见场景说明。

一、系统基础环境（必须）

以 CentOS 7 / 8 为例：

# 更新系统
sudo yum update -y

# 安装基础编译工具
sudo yum install -y gcc gcc-c++ make

# 安装开发依赖
sudo yum install -y kernel-devel

如果你是 CentOS 7，建议 GCC 版本 ≥ 7（默认是 4.8，太老）：

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

二、Python 环境（必须）

CentOS 自带 Python 2/3，但不建议直接用系统 Python。

推荐方式：使用 Miniconda

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

创建环境：

conda create -n torch python=3.9
conda activate torch

三、PyTorch 安装

1️⃣ CPU 版本（无 GPU）

pip install torch torchvision torchaudio

2️⃣ GPU 版本（CUDA）

先确认显卡驱动：

nvidia-smi

安装示例（CUDA 11.8）：

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

四、GPU / CUDA 相关环境（仅 GPU 需要）

1️⃣ NVIDIA 驱动

CentOS 安装驱动示例：

sudo yum install -y epel-release
sudo yum groupinstall -y "Development Tools"

然后安装官方 .run 驱动（推荐）

2️⃣ CUDA Toolkit（通常 PyTorch 自带）

如果自己编译扩展：

sudo yum install -y cuda-toolkit-11-8

五、常见额外依赖

pip install numpy pandas matplotlib scikit-learn

六、最小化环境清单总结

✅ CentOS 7/8
✅ Python 3.8–3.11（Conda 最佳）
✅ GCC ≥ 7
✅ NVIDIA 驱动（GPU）
✅ PyTorch（CPU / CUDA）

如果你告诉我：

• CentOS 版本（7 还是 8）
• 是否有 GPU
• 是否要训练还是只推理

我可以直接给你一条命令可跑的环境配置方案。