AWS Deep Learning Containers for PyTorch Training 2.10 on SageMaker¶
AWS Deep Learning Containers for SageMaker are now available with PyTorch 2.10.
Announcements¶
-
Introduced containers for PyTorch 2.10 for Training on SageMaker
-
Added Python 3.13 support
-
Added PyTorch domain libraries: torchtnt 0.2.4, torchdata 0.11.0, torchaudio 2.10.0, torchvision 0.25.0
-
Added Ubuntu 22.04 support
Core Packages¶
| Package | Version |
|---|---|
| Python | 3.13.11 |
| PyTorch | 2.10.0 |
| TorchVision | 0.25.0 |
| TorchAudio | 2.10.0 |
| TorchData | 0.11.0 |
| TorchTNT | 0.2.4 |
Security Advisory¶
AWS recommends that customers monitor critical security updates in the AWS Security Bulletin.
Reference¶
Docker Image URIs¶
763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:2.10.0-cpu-py313-ubuntu22.04-sagemaker
763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:2.10-cpu-py313-ubuntu22.04-sagemaker-v1
763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:2.10.0-cpu-py313
763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:2.10-cpu-py313
public.ecr.aws/deep-learning-containers/pytorch-training:2.10.0-cpu-py313-ubuntu22.04-sagemaker
public.ecr.aws/deep-learning-containers/pytorch-training:2.10-cpu-py313-ubuntu22.04-sagemaker-v1
public.ecr.aws/deep-learning-containers/pytorch-training:2.10.0-cpu-py313
public.ecr.aws/deep-learning-containers/pytorch-training:2.10-cpu-py313