PyTorch在Linux上的依赖库主要分为系统级和Python包两类,具体如下:
-
系统级依赖:
- Python:推荐3.6+版本。
- pip:Python包管理工具。
- CUDA Toolkit(GPU加速需安装):NVIDIA GPU驱动支持库。
- cuDNN(GPU加速需安装):深度神经网络加速库,需与CUDA版本匹配。
- GCC:GNU编译器集合,用于编译Python扩展。
-
Python包依赖:
- PyTorch核心包:
torch
、torchvision
、torchaudio
。
- 可选扩展包:根据项目需求安装,如
numpy
、pandas
、matplotlib
等。
说明:
- 若使用CPU版本,无需安装CUDA和cuDNN。
- 推荐通过
conda
或pip
安装预编译包,避免手动编译依赖。
- 具体版本需参考PyTorch官方文档,确保与系统环境兼容。