深度学习 AMI
开发人员指南
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 Amazon AWS 入门

CUDA 安装和框架绑定

深度学习都非常一流,但每个框架都提供“稳定”版本。这些稳定版本可能不适用于最新的 CUDA 或 cuDNN 实施和功能。您决定怎么做?这最终取决于您的使用案例和您需要的功能。如果您不确定,则使用最新的 采用 Conda 的 Deep Learning AMI. 它具有所有框架的官方 PIP 二进制文件 ( 包含 CUDA 8 和 CUDA 9),使用每个框架都支持的任何最新版本。

如需进一步指导,请在 稳定与前沿 上查看我们的指南。

CUDA 支持

采用源代码的 Deep Learning AMI 的 CUDA 版本以及每个所支持的框架:

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

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

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

后续步骤

DLAMI 操作系统选项