The AWS Deep Learning AMIs for Ubuntu and Amazon Linux now come with TensorFlow 1.7, which is built with advanced optimizations for high-performance training across Amazon EC2 instance families. This is an update to the optimized build of TensorFlow 1.6 that we launched in late March.

The Amazon Machine Images (AMIs) now come with TensorFlow 1.7 built with Intel’s Advanced Vector Instructions (AVX), SSE, and FMA instruction sets. The AMIs are fully configured with Intel’s Math Kernel Library for Deep Neural Networks (Intel MKL-DNN) to speed up training performance on Intel Xeon Platinum processors powering Amazon EC2 C5 instances.

Training a ResNet-50 benchmark with a synthetic ImageNet dataset using our optimized build of TensorFlow 1.7 on a C5.18xlarge instance type was 9.8X faster than training on the stock TensorFlow 1.7 binaries. The AMIs also come with an optimized build of TensorFlow 1.7 fully configured with NVIDIA CUDA 9 and cuDNN 7 to take advantage of mixed precision training on the Volta V100 GPUs powering Amazon EC2 P3 instances.

A Deep Learning AMI automatically deploys the high performance build of TensorFlow optimized for the EC2 instance of your choice when you activate the TensorFlow virtual environment for the first time. The AMIs also include TensorBoard 1.7 to help you visualize and debug your model training, and TensorFlow Serving 1.6 to quickly prototype an inference endpoint for your trained models.

The Deep Learning AMIs now come with the Microsoft Cognitive Toolkit 2.5 with performance improvements and bug fixes. The AMIs also come with the latest in deep learning frameworks. Read more from…

thumbnail courtesy of