AWS Machine Learning Blog

Optimized TensorFlow 1.8 now available in the AWS Deep Learning AMIs to accelerate training on Amazon EC2 C5 and P3 instances

The AWS Deep Learning AMIs for Ubuntu and Amazon Linux now come with advanced optimizations for TensorFlow 1.8 to deliver higher-performance training for Amazon EC2 C5 and P3 instances.

For CPU-based training scenarios, the Amazon Machine Images (AMIs) now include TensorFlow 1.8, built with Intel’s Advanced Vector Instructions (AVX), SSE, and FMA instruction sets to accelerate vector and floating-point computations. The AMIs are also fully configured with Intel MKL-DNN to accelerate math routines used in neural network training on Amazon EC2 C5 instances. Training a ResNet-50 benchmark with the ImageNet dataset was 7X faster than training on the stock TensorFlow 1.8 binaries when we used an optimized build on a c5.18xlarge instance type with a batch size of 32.

In addition, to improve training performance for GPU-based scenarios, the AMIs include an optimized build of TensorFlow 1.8 fully configured with NVIDIA CUDA 9 and cuDNN 7 to take advantage of mixed precision training on Volta V100 GPUs powering Amazon EC2 P3 instances.

The Deep Learning AMIs automatically deploy higher-performance builds of TensorFlow as well as other deep learning frameworks such as Chainer and CNTK, optimized for the EC2 instance of your choice, when you activate the framework’s virtual environment for the first time.

You can get started with the AWS Deep Learning AMIs by using our quick getting started tutorial and our developer guide for more tutorials and resources. You can also subscribe to our discussion forum to get launch announcements and post your questions.

About the author

Cynthya Peranandam is a Principal Marketing Manager for AWS artificial intelligence solutions, helping customers use deep learning to provide business value. In her spare time she likes to run and listen to music.