PyTorch Deep Learning Containers - Amazon Deep Learning Containers
Services or capabilities described in Amazon Web Services documentation might vary by Region. To see the differences applicable to the China Regions, see Getting Started with Amazon Web Services in China (PDF).

PyTorch Deep Learning Containers

Below are the release notes for PyTorch Deep Learning Containers:

Version Type Service Architecture Release Note
2.7 Training EC2 Arm64

Amazon Deep Learning Containers for PyTorch 2.7 ARM64 (Training on EC2): June 4, 2025
2.6 Training EC2 ECS EKS X86

Amazon Deep Learning Containers for PyTorch 2.6 (Training on EC2, ECS, and EKS): March 12, 2025
2.6 Training SageMaker X86

Amazon Deep Learning Containers for PyTorch 2.6 (Training on SageMaker AI): March 12, 2025
2.6 Inference EC2 ECS EKS X86

Amazon Deep Learning Containers for PyTorch 2.6 (Inference on EC2, ECS, and EKS): February 27, 2025
2.6 Inference SageMaker X86

Amazon Deep Learning Containers for PyTorch 2.6 (Inference on SageMaker AI): February 27, 2025
2.6 Inference EC2 ECS EKS Arm64

Amazon Deep Learning Containers for PyTorch 2.6 ARM64 (Inference on EC2, ECS, and EKS): February 24, 2025
2.6 Inference SageMaker Arm64

Amazon Deep Learning Containers for PyTorch 2.6 ARM64 (Inference on SageMaker AI): February 24, 2024
2.5 Training EC2 ECS EKS X86

Amazon Deep Learning Containers for PyTorch 2.5 (Training on EC2, ECS, and EKS): November 25, 2024
2.5 Training SageMaker X86

Amazon Deep Learning Containers for PyTorch 2.5 (Training on SageMaker AI): November 25, 2024
2.5 Inference EC2 ECS EKS X86

Amazon Deep Learning Containers for PyTorch 2.5 (Inference on EC2, ECS, and EKS): November 13, 2024
2.5 Inference SageMaker X86

Amazon Deep Learning Containers for PyTorch 2.5 (Inference on SageMaker AI): November 13, 2024
2.5 Inference EC2 ECS EKS Arm64

Amazon Deep Learning Containers for PyTorch 2.5 ARM64 (Inference on EC2, ECS, and EKS): December 11, 2024
2.5 Inference SageMaker Arm64

Amazon Deep Learning Containers for PyTorch 2.5 ARM64 (Inference on SageMaker AI): December 11, 2024
2.4 Training EC2 ECS EKS X86

Amazon Deep Learning Containers for PyTorch 2.4 (Training on EC2, ECS, and EKS): September 26, 2024
2.4 Training SageMaker X86

Amazon Deep Learning Containers for PyTorch 2.4 (Training on SageMaker AI): September 26, 2024
2.4 Inference EC2 ECS EKS X86

Amazon Deep Learning Containers for PyTorch 2.4 (Inference on EC2, ECS, and EKS): September 26, 2024
2.4 Inference SageMaker X86

Amazon Deep Learning Containers for PyTorch 2.4 (Inference on SageMaker AI): October 3, 2024
2.4 Inference EC2 ECS EKS Arm64

Amazon Deep Learning Containers for PyTorch 2.4 Graviton (Inference on EC2, ECS, and EKS): July 11, 2024
2.4 Inference SageMaker Arm64

Amazon Deep Learning Containers for PyTorch 2.4 Graviton (Inference on SageMaker AI): July 11, 2024