CentOS系统中PyTorch的安装步骤

CentOS 安装 PyTorch 步骤

一 准备环境

• 更新系统并安装基础工具
  • 执行：sudo yum update -y
  • 安装编译与基础依赖：sudo yum groupinstall -y "Development Tools" 以及 sudo yum install -y cmake3 git wget
• 安装 Python 3 与 pip
  • 执行：sudo yum install -y python3 python3-pip python3-devel
• 建议创建隔离环境（任选其一）
  • 使用 venv：python3 -m venv pytorch-env && source pytorch-env/bin/activate
  • 使用 conda：conda create -n pytorch python=3.8 然后 conda activate pytorch
• 说明：常见可用 Python 版本为 3.6+，虚拟环境可避免包冲突并便于管理依赖。

二 选择安装方式

• CPU 版本（最简便，适合无 NVIDIA GPU 或仅做推理/CPU 训练）
  • 在激活的虚拟环境中执行：pip install torch torchvision torchaudio
• GPU 版本（需已安装匹配版本的 NVIDIA 驱动、CUDA Toolkit 与 cuDNN）
  • 先确认 CUDA 版本：nvcc --version
  • 按 CUDA 版本选择安装命令（示例为 CUDA 11.7）：pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
  • 若使用 conda，可用：conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
• 版本匹配要点
  • PyTorch 版本需与 CUDA/cuDNN 兼容；不确定时优先使用 CPU 版本或参考 PyTorch 官方安装页面生成命令。

三 验证安装

• 在 Python 中执行：
  • import torch
  • print(torch.__version__)
  • print(torch.cuda.is_available())
• 预期结果
  • 输出版本号；如安装的是 GPU 版本且驱动/库配置正确，torch.cuda.is_available() 返回 True。

四 常见问题与处理

• 找不到命令或包：确认已激活虚拟环境，并使用 python3 -m pip 确保安装到当前环境
• 旧版 GCC 导致编译失败：升级 GCC（推荐 ≥7.3.0）；若安装 PyTorch 1.11.0，建议 GCC ≥7.5.0
• 需要编译场景的依赖：安装 patch、libjpeg-turbo-devel、dos2unix、openblas、git、cmake 等系统依赖
• 需要离线安装：从 PyTorch 官方 whl 页面下载对应版本（如 torch-1.8.1+cpu-cp37-cp37m-linux_x86_64.whl），然后 pip3 install 文件名.whl
• 需要加速下载：可使用国内镜像源（如清华源）进行 pip 安装（注意与 PyTorch 官方索引的兼容性）。