温馨提示×

CentOS PyTorch依赖库有哪些

小樊
43
2025-12-17 19:55:58
栏目: 智能运维

CentOS 上 PyTorch 的依赖库与版本要求

系统级依赖

  • 基础工具与构建链:Development Tools(包含 gcc、make 等)、cmake3gitwget
  • Python 运行与开发:python3python3-develpython3-pip
  • 常见图像处理依赖:patchlibjpeg-turbo-develdos2unix
  • CPU 线性代数库:openblas(或系统可用的 BLAS/LAPACK 实现)。
    以上为在 CentOS 上通过源码编译或使用官方预编译包时常见的系统级依赖项。

GPU 相关依赖(可选)

  • CUDA Toolkit:需与所选 PyTorch 版本匹配(如 CUDA 11.x 等)。
  • cuDNN:与 CUDA 版本兼容的深度学习加速库。
  • 环境配置:安装后设置 PATHLD_LIBRARY_PATH,例如将 /usr/local/cuda/bin 加入 PATH,将 /usr/local/cuda/lib64 加入 LD_LIBRARY_PATH。
    若不使用 GPU,可仅安装 CPU 版本,无需上述组件。

Python 依赖

  • 核心数值库:numpy
  • 科学计算与数据处理:scipypandas
  • 可视化:matplotlib
    这些为深度学习开发与训练常用的配套库,建议与 PyTorch 一并安装。

版本要求与注意事项

  • 编译器与构建工具:建议 gcc ≥ 7.3.0cmake ≥ 3.12.0;若安装 PyTorch 1.11.0,建议 gcc ≥ 7.5.0
  • Python 版本:建议使用 Python 3.6+
  • 验证安装:执行 python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())",确认版本与 GPU 可用性。

0