The AWS Deep Learning AMIs for Ubuntu and Amazon Linux now come with advanced optimizations tailored for TensorFlow 1.6 to deliver high-performance training across Amazon EC2 instance families.  

Source: Amazon Web Services News AWS Deep Learning AMIs Now Include Optimized TensorFlow 1.6 for Amazon EC2 P3 and C5 Instances