一聚教程网:一个值得你收藏的教程网站

最新下载

热门教程

CentOS PyTorch依赖库有哪些

时间:2026-07-09 09:34:45 编辑:袖梨 来源:一聚教程网

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

CentOS PyTorch依赖库有哪些

系统级依赖

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

GPU 相关依赖(可选)

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

Python 依赖

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

版本要求与注意事项

  • 编译器与构建工具:建议 gcc ≥ 7.3.0、cmake ≥ 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 可用性。

热门栏目