亚马逊AWS官方博客

AWS Deep Learning AMI 现在能够以更快的速度训练适用于 TensorFlow 和 Microsoft Cognitive Toolkit 的 Volta GPU

现在,适用于 UbuntuAmazon Linux 的 AWS Deep Learning AMI 包含最新版本的 TensorFlow (1.5) 和 Microsoft Cognitive Toolkit (2.4)。这些框架支持 NVIDIA CUDA 9 和 cuDNN 7 驱动程序。这可让您利用 V100 Volta GPU (支持 Amazon EC2 P3 实例) 支持的混合精度训练。在早期在 Volta 上进行的 TensorFlow 1.5 测试中,我们在 p3.8xlarge 实例上使用 ImageNet 合成数据在 FP16 模式下对 ResNet-50 基准进行训练,其速度比使用 TensorFlow 1.4.1 进行训练要快 1.8 倍

深度学习框架最新动态

Deep Learning AMI 提供预置的 pip 二进制文件,适用于基于 Conda 的不同虚拟环境中的深度学习框架最新官方版本。每个框架都预先配置了框架支持的 NVIDIA CUDA 最新版本。

使用 CUDA 9 的框架:

  • Apache MXNet 1.0 (及 Gluon)
  • Caffe2 0.8.1
  • Microsoft Cognitive Toolkit (CNTK) 2.4
  • PyTorch 0.3
  • TensorFlow 1.5
  • Theano 1.0
  • 使用 CUDA 8 的 Caffe 1.0
  • Keras 1.2.2 和 Keras 2.1.3

这些 AMI 还包含以下工具提供的模型应用和调试功能:

  • 适用于 Apache MXNet 的模型服务器 0.1
  • TensorFlow Serving 1.4.0
  • TensorBoard 1.0.0

开始使用 AWS Deep Learning AMI

您可在 AWS Marketplace 获取最新版本的 AWS Deep Learning AMI。我们的 AMI 选择主题有助于您选取适合您深度学习项目的 AMI。此外,我们还提供了大量的教程和开发人员资源,可帮助您快速在 AWS 上部署第一个深度学习模型。


Sumit Thakur 是 AWS 深度学习高级产品经理。他负责开发让客户能够轻松开始在云端进行深度学习的产品,他专注于使 Deep Learning AMI 引擎更易用。在业余时间,他喜欢亲近自然和观看科幻电视剧。