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