PyTorch Deep Learning Containers - AWS Deep Learning Containers

PyTorch Deep Learning Containers

Below are the release notes for PyTorch Deep Learning Containers:

Version Type Service Architecture Release Note
2.9 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.9 (Training on SageMaker AI): December 12, 2025

2.9 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.9 (Training on EC2, ECS, and EKS): December 12, 2025

2.8 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.8 (Training on SageMaker AI): September 13, 2025

2.8 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.8 (Training on EC2, ECS, and EKS): September 13, 2025

2.7 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.7 (Training on SageMaker AI): June 13, 2025

2.7 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.7 (Training on EC2, ECS, and EKS): June 13, 2025

2.7 Training EC2 Arm64

AWS Deep Learning Containers for PyTorch 2.7 ARM64 (Training on EC2): June 4, 2025

2.6 Training EC2 ECS EKS X86

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

2.6 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.6 (Training on SageMaker AI): March 12, 2025

2.6 Inference EC2 ECS EKS X86

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

2.6 Inference SageMaker X86

AWS Deep Learning Containers for PyTorch 2.6 (Inference on SageMaker AI): February 27, 2025

2.6 Inference EC2 ECS EKS Arm64

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

2.6 Inference SageMaker Arm64

AWS Deep Learning Containers for PyTorch 2.6 ARM64 (Inference on SageMaker AI): February 24, 2024