Deep Learning AMI
Developer Guide

AWS Deep Learning Containers on Amazon SageMaker

AWS Deep Learning Containers are a set of Docker images for training and serving models in TensorFlow, PyTorch, and MXNet on Amazon SageMaker. AWS Deep Learning Containers provide optimized environments with TensorFlow and MXNet, Nvidia CUDA (for GPU instances), and Intel MKL (for CPU instances) libraries and are available in Amazon ECR. For more information, see the SageMaker Documentation