You can now use TensorFlow 1.5 and Apache MXNet 1.0 with the pre-built deep learning framework containers that are provided with Amazon SageMaker. Amazon SageMaker also now supports CUDA 9 drivers, for faster performance on Amazon SageMaker ml.p3 instances. Both updates to TensorFlow and MXNet provide improved performance features, among other details that can be found on the respective GitHub repositories for TensorFlow 1.5 and MXNet 1.0.0. 

Source: Amazon Web Services News Amazon SageMaker now supports TensorFlow 1.5, Apache MXNet 1.0, and CUDA 9 for P3 Instance Optimization