温馨提示×

Ubuntu下PyTorch的依赖库有哪些

小樊
53
2025-08-17 22:46:07
栏目: 智能运维

在Ubuntu下安装PyTorch,通常需要以下依赖库:

  • Python相关:Python 3.6及以上版本、pip(Python包管理工具)、setuptools(打包分发工具)、wheel(二进制包格式,加快安装)。
  • CUDA相关(可选,用于GPU加速):CUDA Toolkit、cuDNN(与CUDA配套,优化深度学习模型性能)、NVIDIA驱动程序(需与CUDA版本兼容)。
  • 其他:build-essential(包含编译工具和库,用于编译安装Python扩展模块)。

若从源码编译,还需安装git(版本控制工具)。

0