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

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

CUDA 安装和框架绑定

尽管深度学习都非常先进,但每个框架都提供 “稳定” 版本。这些稳定版本可能不适用于最新的 CUDA 或 cuDNN 实施和功能。您的使用案例和您需要的功能可以帮助您选择框架。如果你不确定,那么将最新的深度学习 AMI 与 Conda 结合使用。它有官方pip所有框架的二进制文件 (包含 CUDA 10),使用每个框架都支持的任何最新版本。如果您需要最新版本并自定义深度学习环境,请使用深度学习库 AMI。

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

用 CUDA 选择 DLAMI

这些区域有:AMI 深度学习基础拥有所有可用的 CUDA 11 系列,包括 11.0、11.1 和 11.2。

这些区域有:使用 Conda 进行深度学习 AMI拥有所有可用的 CUDA 11 系列,包括 11.0、11.1 和 11.2。

注意

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

有关特定框架版本号,请参阅DLAMI 的发布说明

选择这个 DLAMI 类型,或通过后续步骤选项。

选择一个 CUDA 版本并在附录,或者使用后续步骤选项。

后续步骤

AMI 深度学习基础