AWS Deep Learning Containers for Ray 2.54 on SageMaker¶
AWS Deep Learning Containers for SageMaker are now available with Ray 2.54.
Announcements¶
-
Introduced Ray Serve 2.54 GPU containers for SageMaker
-
Built on Amazon Linux 2023 with Python 3.13
Core Packages¶
| Package | Version |
|---|---|
| Ray Serve | 2.54.1 |
| PyTorch | 2.10.0 |
| TorchVision | 0.25.0 |
| TorchAudio | 2.10.0 |
| Transformers | 5.2.0 |
| scikit-learn | 1.8.0 |
| FFmpeg | 8.0.1 |
| CUDA | 12.9.1 |
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/ray:serve-ml-sagemaker-cuda-v1.0.0
public.ecr.aws/deep-learning-containers/ray:serve-ml-sagemaker-cuda-v1.0.0