AWS Deep Learning Containers for TensorFlow Training 2.19 on SageMaker¶
AWS Deep Learning Containers for SageMaker are now available with TensorFlow 2.19.
Announcements¶
-
Introduced TensorFlow 2.19 training containers for SageMaker
-
Added Python 3.12 support
Core Packages¶
| Package | Version |
|---|---|
| TensorFlow | 2.19.0 |
| Python | 3.12.11 |
| CUDA | 12.5.1 |
| cuDNN | 9.3.0.75 |
| EFA | 1.40.0 |
| OpenMPI | 4.1.8 |
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/tensorflow-training:2.19.0-gpu-py312-cu125-ubuntu22.04-sagemaker
763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-training:2.19-gpu-py312-cu125-ubuntu22.04-sagemaker-v1
763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-training:2.19.0-gpu-py312-sagemaker
763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-training:2.19-gpu-py312-sagemaker
public.ecr.aws/deep-learning-containers/tensorflow-training:2.19.0-gpu-py312-cu125-ubuntu22.04-sagemaker
public.ecr.aws/deep-learning-containers/tensorflow-training:2.19-gpu-py312-cu125-ubuntu22.04-sagemaker-v1
public.ecr.aws/deep-learning-containers/tensorflow-training:2.19.0-gpu-py312-sagemaker
public.ecr.aws/deep-learning-containers/tensorflow-training:2.19-gpu-py312-sagemaker