温馨提示×

PyTorch在Linux上的依赖库有哪些

linux

小樊

40

2025-08-29 19:26:25

栏目: 智能运维

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官方文档，确保与系统环境兼容。

0 赞

0 踩

最新问答

相关问答

相关标签

产品服务

地区划分

专题活动

帮助支持

关于我们

售后咨询

7*24小时在线电话：400-100-2938

7*24小时在线 QQ：800811969

关注亿速云

亿速云公众号

手机网站二维码