CUDA 安装和框架绑定 - 深度学习 AMI
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 AWS 服务入门

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

CUDA 安装和框架绑定

深度学习都非常一流,但每个框架都提供“稳定”版本。这些稳定版本可能无法与最新的CUDA或 cuDNN 实施和功能。您决定怎么做? 这最终取决于您的使用案例和您需要的功能。如果您不确定,则使用最新的 采用 Conda 的 Deep Learning AMI. 它具有所有框架的正式pip二进制文件,CUDA10使用每个框架支持的最新版本。如果您需要最新版本,并要自定义深度学习环境,请使用 Deep Learning Base AMI。

如需进一步指导,请在 稳定性与候选版本 上查看我们的指南。

选择具有 CUDA 的 DLAMI

Deep Learning Base AMI 有CUDA10、10.1和10.2。

采用 Conda 的 Deep Learning AMI 有CUDA10、10.1和10.2。

  • CUDA10.1配 cuDNN 七: Apache(Apa MXNet (孵育), PyTorch,和 TensorFlow 2间

  • CUDA10配 cuDNN 七: TensorFlow 和链工

注意

从 v28 版本开始,AWS Deep Learning AMI 中将不再包含 CNTK、Caffe、Caffe2 和 Theano Conda 环境。包含这些环境的先前版本的 AWS Deep Learning AMI 将继续可用。但是,只有在开源社区针对这些框架发布安全修补程序时,我们才会为这些环境提供更新。

有关 DLAMI 类型和操作系统的安装选项,请参阅每个 CUDA 版本和 选项页面:

可以在 DLAMI 的发行说明中找到特定的框架版本号

选择这个 DLAMI 类型,或通过“后续步骤”选项了解有关其他 DLAMI 的更多信息。

选择一个 CUDA 版本并在附录中查看具有该版本的完整 DLAMI 列表,或者使用“后续步骤”选项了解不同 DLAMI 的更多信息。

后续步骤

DLAMI 操作系统选项

相关主题