The latest release of the AWS Deep Learning AMI is available for Ubuntu and Amazon Linux platforms. The AMI includes several updates, the most notable of which is MXNet 0.11 with Keras Support. MXNet is now available as a backend choice for users running Keras 1.2. MXNet gives you a simple way to set up multiple GPUs for training and superior performance that scales near-linearly. To read more about the benefits of using MXNet as a backend for Keras, refer to NVIDIA’s blog. The AMI also now runs with latest version of TensorFlow, which is version 1.3.0.
Here is the complete list of deep learning frameworks and GPU drivers on the AMI (both Ubuntu and Amazon Linux versions):
Keras 1.2.2 (DMLC fork with MXNet 0.11 support)
Torch (master branch)CUDA Toolkit 8.0
NVIDIA Driver 375.66