Posted On: Apr 30, 2018

The AWS Deep Learning AMIs now include advanced optimizations for TensorFlow 1.7 that are tailored to deliver higher-performance training across Amazon EC2 instances.

For CPU-based training scenarios, the AMIs now include TensorFlow 1.7 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 Intel Xeon processors powering Amazon EC2 C3 and C4 instances. 

In addition, to improve training performance for GPU-based scenarios, the AMIs include an optimized build of TensorFlow 1.7 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 AMIs automatically deploy the build of TensorFlow optimized for the EC2 instance of your choice when you activate the TensorFlow virtual environment for the first time.

You can find the Deep Learning AMI of your choice in the Quick Start section of the Step 1: Choose an Amazon Machine Image (AMI) in the EC2 instance launch wizard. Also visit our Documentation guide for help with selecting the right AMI for your project, simple tutorials and more resources.