AWS 深度学习 AMI 现在包括 Apache MXNet 0.11 和 TensorFlow 1.3.0
AWS 深度学习亚马逊机器映像 (AMI) 旨在帮助您在 AWS 上构建稳定、安全和可扩展的深度学习应用程序。AMI 预装了流行的深度学习框架。它具有 GPU 驱动程序和库,可让您训练复杂的 AI 模型并在云端对其进行扩展。
最新版本的 AWS 深度学习 AMI 可用于 Ubuntu 和亚马逊 Linux 平台。AMI 包含多项更新,其中最值得注意的是支持 Keras 的 MXNet 0.11。运行 Keras 1.2 的用户现在可以选择 MXNet 作为后端选项。MXNet 为您提供了一种设置多个 GPU 以进行训练和近线性扩展的卓越性能的简单方法。要详细了解使用 MXNet 作为 Keras 后端的好处,请参阅 NVIDIA 的博客。AMI 现在还使用最新版本的 TensorFlow 运行,即 1.3.0 版。
开始使用 AMI 既快速又简单。按照此分步指南进行操作。
以下是 AMI(Ubuntu 和亚马逊 Linux 版本)上的深度学习框架和 GPU 驱动程序的完整列表:
MXNet 0.11.0
TensorFlow 1.3.0
Keras 1.2.2(支持 MXNet 0.11 的 DMLC 分支)
Caffe1.0
Caffe2 0.8.0
CNTK 2.0
Theano 0.9.0
Torch(主分支)CUDA 工具包 8.0
cuDNN 5.1
NVIDIA 驱动程序 375.66
你可以在 AWS Marketplace 上找到适用于 Ubuntu 和亚马逊 Linux 的 AMI 。