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 .