CentOS PyTorch版本升级指南

CentOS 上 PyTorch 版本更新指南

一 准备与环境检查

• 建议在独立的虚拟环境中操作，避免影响系统或其他项目：使用 conda create -n torch_env python=3.9 创建环境，随后 conda activate torch_env。
• 升级包管理工具：pip install --upgrade pip。
• 查看当前 PyTorch 版本与位置：pip show torch；如需查看可升级包：pip list --outdated --format=legacy。
• 如需 GPU 支持，先确认驱动与 CUDA：运行 nvidia-smi 检查驱动与 CUDA 运行时版本，确保与待安装的 PyTorch 版本兼容。

二 使用 pip 更新

• 升级到最新可用版本（适合大多数场景）：pip install --upgrade torch torchvision torchaudio。
• 升级到指定版本（锁定版本更可控）：pip install torch== torchvision== torchaudio==。
• 指定 CPU 或特定 CUDA 版本（示例为 CUDA 11.7）：pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117；CPU 版本可直接 pip install torch torchvision torchaudio。
• 遇到冲突或不干净状态时，可先卸载再装：pip uninstall torch torchvision torchaudio。

三 使用 conda 更新

• 激活环境：conda activate your_env。
• 更新到最新兼容版本：conda update torch torchvision torchaudio。
• 指定 CUDA 工具链版本（示例为 11.3）：conda update torch torchvision torchaudio cudatoolkit=11.3 -c pytorch；若仅 CPU：conda install pytorch torchvision torchaudio cpuonly -c pytorch。
• 指定具体版本：conda install pytorch= torchvision= torchaudio= -c pytorch。

四 验证与常见问题处理

• 版本与设备可用性验证：
  • 版本：python -c “import torch; print(torch.version)”
  • CUDA 可用性：python -c “import torch; print(torch.cuda.is_available())”
  • PyTorch 构建的 CUDA 版本：python -c “import torch; print(torch.version.cuda)”
• 常见问题与要点：
  • 升级前先备份环境与数据，避免依赖冲突导致项目不可用。
  • 升级后运行单元测试或关键样例，确认数值一致性与性能符合预期。
  • GPU 环境需确保 NVIDIA 驱动 与 CUDA/cuDNN 与 PyTorch 版本匹配；如不确定，优先使用 conda 的 cudatoolkit 管理 CUDA 运行时。
  • 若遇到 pip/conda 解析冲突，优先在干净虚拟环境中重装；必要时使用 pip uninstall 清理后再安装。

五 特殊场景与进阶

• 从源码编译（仅在确有需要时采用，如定制优化或特殊架构）：
  • 安装系统依赖（CentOS 示例）：yum install -y patch libjpeg-turbo-devel dos2unix openblas git；编译器要求 gcc≥7.3.0、cmake≥3.12.0（部分版本如 1.11.0 建议 gcc≥7.5.0）。
  • 快速安装 CPU 版本（示例）：pip3 install torch==1.8.1+cpu；或下载对应 whl 安装。
  • 源码编译流程要点：git clone -b https://github.com/pytorch/pytorch.git --depth=1，随后 git submodule sync && git submodule update --init --recursive，按需设置环境变量并执行 python3 setup.py install。
  • 注意选择与 PyTorch 版本匹配的 torchvision（如 1.8.1→0.9.1、1.11.0→0.12.0、2.0.1→0.15.2）。