亚马逊AWS官方博客

AWS Deep Learning AMI 现在包含 Apache MXNet 0.11 和 TensorFlow 1.3

AWS Deep Learning Amazon 系统映像 (AMI) 旨在帮助您在 AWS 上构建稳定、安全且可扩展的深度学习应用程序。AMI 预安装了常用的深度学习框架,具有 GPU 驱动程序和库,让您可以训练复杂的 AI 模型并在云中进行扩展。

最新版 AWS Deep Learning AMI 可用于 Ubuntu 和 Amazon Linux 平台。AMI 中包括多项更新,其中最引人注目的是支持 Keras 的 MXNet 0.11。MXNet 现可作为运行 Keras 1.2 的用户的后端选择。利用 MXNet,Keras 用户可以轻松设置多 GPU 以实现训练和近线性扩展的出色性能。要了解有关使用 MXNet 作为 Keras 后端的好处的更多信息,请参阅 NVIDIA 的博客。AMI 现在还可与最新版 TensorFlow (1.3.0 版) 一起运行。

使用 AMI 进行入门可谓轻松快捷。按照此分步指南 进行操作,在几分钟内便可开始深度学习。

以下是 AMI 上的深度学习框架和 GPU 驱动程序的完整列表 (同时适用于 Ubuntu 和 Amazon Linux 版本):

  • MXNet 0.11.0
  • TensorFlow 1.3.0
  • Keras 1.2.2 (DMLC 复制,具有 MXNet 0.11 支持)
  • Caffe1.0
  • Caffe2 0.8.0
  • CNTK 2.0
  • Theano 0.9.0
  • Torch (主分支)
  • CUDA Toolkit 8.0
  • cuDNN 5.1
  • NVIDIA Driver 375.66

您可以在 AWS Marketplace 上找到 AMI for UbuntuAmazon Linux