本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
CUDA 安装和框架绑定
虽然深度学习都非常先进,但每个框架都提供“稳定”版本。这些稳定版本可能不适用于最新的 CUDA 或 cuDNN 实施和功能。您的用例和所需功能可以帮助您选择框架。如果您不确定,就请使用最新的带 Conda 的深度学习 AMI。它具有适用于所有框架的官方 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。
注意
从 v28 版本开始,我们将不再在 Amazon Deep Learning AMI 中包含 CNTK、Caffe、Caffe2、Theano、Chainer 或 Keras Conda 环境。包含这些环境的先前版本的 Amazon Deep Learning AMI 继续可用。但是,只有在开源社区针对这些框架发布安全修补程序时,我们才为这些环境提供更新。
有关具体框架版本号,请参阅 DLAMI 的发布说明。
选择这个 DLAMI 类型,或通过后续步骤选项了解有关其他 DLAMI 的更多信息。
选择一个 CUDA 版本并在附录中查看具有该版本的完整 DLAMI 列表,或者使用后续步骤选项来了解不同 DLAMI 的更多信息。
后续步骤
相关主题
-
有关在 CUDA 版本之间切换的说明,请参阅使用深度学习基础 AMI教程。