Deep Learning AMI
Developer Guide

AWS Deep Learning Containers on Amazon EKS

AWS Deep Learning Containers are a set of Docker images for training and serving models in TensorFlow, PyTorch, and MXNet on EKS and Elastic Container Service (Amazon ECS). Deep Learning Containers provide optimized environments with TensorFlow, PyTorch, and MXNet, Nvidia CUDA (for GPU instances), and Intel MKL (for CPU instances) libraries and are available in the Elastic Container Registry(Amazon ECR).

This guide describes how to use AWS Deep Learning Containers with EKS.