发布于: Oct 27, 2017
我们非常高兴地宣布推出一套新的 AWS 深度学习 AMI,它们预装了针对新的 Amazon EC2 P3 实例系列中的 NVIDIA Volta V100 GPU 优化的深度学习框架。
新 AMI 为 Ubuntu 和 Amazon Linux 预装并配置了 CUDA 9 及其他 GPU 驱动程序 (包括 cuDNN 7.0、NCCL 2.0.5 和 NVIDIA 驱动程序 384.81),以便充分发挥 P3 上 Volta 的速度优势。
除了最新的驱动程序以外,这些 AMI 还包含为 P3 和 Volta 优化过的常见框架,包括 Apache MXNet v0.12 RC1、Caffe2 v0.8.1 和 Tensorflow (master)。
AWS 深度学习 AMI 入门很简单。只需在 AWS Marketplace 中单击一下鼠标就能开始;或者,您也可以按照此分步指南操作,开始使用您的第一个笔记本。
CUDA 9 AMI 提供以下版本:
当有其他框架支持 Volta 时,我们会尽快加入它们。我们仍维护和提供 CUDA 8 AMI,并整合 TensorFlow、MXNet、Caffe、Caffe 2、Microsoft Cognitive Toolkit (CNTK) 及 Keras 的最新稳定、官方修复版本。CUDA 8 AMI 提供以下版本: