AWS Deep Learning AMI 现已包含 Apache MXNet 0.11 和 TensorFlow 1.3.0

发布于: 2017年9月18日

AWS Deep Learning Amazon Machine Image(AMI)旨在帮助您在 AWS 上构建稳定、安全、可扩展的深度学习应用程序。AMI 已预装常用的深度学习框架。它配备 GPU 驱动程序和库,可让您训练复杂的 AI 模型,并在云中进行扩展。 

AWS Deep Learning AMI 的最新版本可用于 Ubuntu 和 Amazon Linux 平台。AMI 包含多项更新,其中最值得注意的是包含 Keras 支持的 MXNet 0.11。对于运行 Keras 1.2 的用户,现可将 MXNet 作为后端选项使用。MXNet 为您提供一种设置多个 GPU 以进行训练和实现近乎线性扩展的卓越性能的简单方法。要详细了解将 MXNet 用作 Keras 后端的优势,请参阅 NVIDIA 的博客。AMI 现也支持最新版本的 TensorFlow,即版本 1.3.0。

开始使用 AMI 既快速又简单。按照此分步指南操作。 

以下是 AMI(包括 Ubuntu 和 Amazon Linux 版本)上所有深度学习框架和 GPU 驱动程序的完整列表: 

MXNet 0.11.0
TensorFlow 1.3.0
Keras 1.2.2(DMLC 分支,支持 MXNet 0.11)
Caffe 1.0
Caffe2 0.8.0
CNTK 2.0
Theano 0.9.0
Torch(主分支)CUDA 工具包 8.0

cuDNN 5.1
NVIDIA Driver 375.66 

您可以通过 AWS Marketplace 查找 适用于 UbuntuAmazon Linux 的 AMI。