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

CUDA 安装和框架绑定

虽然深度学习都非常先进,但每个框架都提供“稳定”版本。这些稳定版本可能不适用于最新的 CUDA 或 cuDNN 实施和功能。您的用例和所需功能可以帮助您选择框架。如果您不确定,就请使用最新的带 Conda 的深度学习 AMI。它为所有使用 CUDA 的框架提供了官方 pip 二进制文件,同时使用每个框架均支持的最新版本。如果您需要最新版本,并要自定义深度学习环境,就请使用深度学习基础 AMI。

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

选择带 CUDA 的 DLAMI

深度学习基础 AMI 具有所有可用的 CUDA 版本系列

带 Conda 的深度学习 AMI 具有所有可用的 CUDA 版本系列

注意

我们不再在 Amazon Deep Learning AMIs 中包含 MXNet、CNTK、Caffe、Caffe2、Theano、Chainer 或 Keras Conda 环境。

有关具体框架版本号,请参阅 Deep Learning AMI 发布说明

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

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

后续步骤

深度学习基础 AMI