centos pytorch依赖什么

在 CentOS 上安装和使用 PyTorch，通常依赖以下几类内容，按“系统依赖 → Python 依赖 → 可选依赖”说明：

一、系统级依赖（CentOS）

1. 基础编译/运行工具

sudo yum groupinstall -y "Development Tools"
sudo yum install -y epel-release
sudo yum install -y \
  gcc gcc-c++ make \
  git wget curl

2. Python 相关

CentOS 7 默认 Python 2，建议用 Python 3：

sudo yum install -y python3 python3-devel python3-pip

3. CPU 版本常见依赖

sudo yum install -y \
  libgomp \
  glibc \
  numactl

4. GPU（CUDA）版本依赖（如用 NVIDIA）

• 正确版本的 NVIDIA 驱动
• CUDA Toolkit
• cuDNN
• 显卡支持（如 Tesla / RTX）

nvidia-smi
nvcc --version

二、Python 依赖（PyTorch 自身）

1. 必需 Python 包

PyTorch 安装时会自动处理：

• numpy
• typing-extensions
• filelock
• sympy
• networkx
• jinja2
• fsspec

2. 常用附加依赖

pip install torch torchvision torchaudio

如只装 CPU 版：

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

三、可选但常见依赖

1. 科学计算 / 数据处理

pip install numpy pandas matplotlib scipy

2. 分布式训练

• nccl
• openmpi

sudo yum install -y openmpi openmpi-devel

3. 深度学习相关

pip install scikit-learn tqdm pillow

四、CentOS 常见注意点

1. CentOS 7

   • glibc 版本较低（2.17）
   • 新 PyTorch（2.x）可能不支持
   • 推荐：CentOS 8 / Rocky / Alma / Ubuntu

2. Python 版本

   • PyTorch 2.x 通常要求 Python ≥ 3.8

3. 离线环境

   • 需提前下载 .whl 包
   • 注意 CUDA / cuDNN 匹配

五、快速检查是否成功

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

如果你能告诉我：

• CentOS 版本（7 / 8）
• CPU 还是 GPU
• Python 版本

我可以给你一套精确的安装命令。