AWS Deep Learning (DL) Containers now support PyTorch. AWS DL Containers are Docker images pre-installed with deep learning frameworks to make it easy to setup and deploy custom machine learning environments. Docker images for training and inference with PyTorch are now available through Amazon Elastic Container Registry (Amazon ECR) free of charge—you pay only for the resources that you use. You can deploy AWS DL Containers on Amazon SageMaker, Amazon Elastic Kubernetes Service (Amazon EKS), self-managed Kubernetes on Amazon EC2, and Amazon Elastic Container Service (Amazon ECS).
Source:: Amazon AWS