Deep Learning Containers Images - AWS Deep Learning Containers

Deep Learning Containers Images

AWS Deep Learning Containers are available as Docker images in Amazon ECR. Each Docker image is built for training or inference on a specific Deep Learning framework version, python version, with CPU or GPU support.

For the full list of available Deep Learning Containers and information on pulling them, see Available Deep Learning Containers Images.

Once you've selected your desired Deep Learning Containers image, continue with the one of the following:

  • To run training and inference on Deep Learning Containers for Amazon EC2 using MXNet, PyTorch, TensorFlow, and TensorFlow 2, see Amazon EC2 Tutorials

  • To run training and inference on Deep Learning Containers for Amazon ECS using MXNet, PyTorch, and TensorFlow, see Amazon ECS tutorials

  • Deep Learning Containers for Amazon EKS offer CPU, GPU, and distributed GPU-based training, as well as CPU and GPU-based inference. To run training and inference on Deep Learning Containers for Amazon EKS using MXNet, PyTorch, and TensorFlow, see Amazon EKS Tutorials

  • For information on security in Deep Learning Containers, see Security in AWS Deep Learning Containers

  • For a list of the latest Deep Learning Containers release notes, see Release Notes for Deep Learning Containers