Skip to content

Releases: aws/deep-learning-containers

v1.1-pt-arm64-ec2-2.6.0-inf-py312

08 Mar 00:23
0162d79
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-arm64:2.6.0-cpu-py312-ubuntu22.04-ec2-v1.1
sha256:e3504cff6ab601654675953323855fdcc55a766f153c324c00f9baab2b62fcf8

All tags:

2.6.0-cpu-py312-ec2
2.6.0-cpu-py312-ubuntu22.04-ec2
2.6.0-cpu-py312-ubuntu22.04-ec2-v1.1
2.6-cpu-py312-ec2
2.6-cpu-py312-ubuntu22.04-ec2-v1
2.6.0-cpu-py312-ubuntu22.04-ec2-v1.1-2025-03-07-20-51-10

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.38.8 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.37.8 libgcc-11-dev:arm64 & libgcc-s1:arm64
botocore==1.37.8 libgcc-11-dev:arm64 & libgcc-s1:arm64
cffi==1.17.1 libstdc++-11-dev:arm64 & libstdc++6:arm64
Cython==3.0.12
filelock==3.17.0
ninja==1.11.1.1
numpy==2.2.3
opencv-python==4.11.0.86
packaging==24.2
pillow==11.1.0
requests==2.32.3
s3transfer==0.11.4
scipy==1.15.2
setuptools==75.8.2
torch==2.6.0+cpu
torch-model-archiver==0.12.0
torchaudio==2.6.0+cpu
torchserve==0.12.0
torchvision==0.21.0+cpu

  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-arm64:2.6.0-gpu-py312-cu124-ubuntu22.04-ec2-v1.1
sha256:9515b10f57154657818d2885d78310a2fbb38c75ed1af7abb5b4dcb5f938078a

All tags:

2.6.0-gpu-py312-cu124-ubuntu22.04-ec2-v1.1-2025-03-07-20-51-21
2.6-gpu-py312-ec2
2.6.0-gpu-py312-cu124-ubuntu22.04-ec2
2.6-gpu-py312-cu124-ubuntu22.04-ec2-v1
2.6.0-gpu-py312-ec2
2.6.0-gpu-py312-cu124-ubuntu22.04-ec2-v1.1

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 cuda-command-line-tools-12-4
awscli==1.38.8 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.37.8 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
botocore==1.37.8 libcublas-12-4 & libcublas-dev-12-4
cffi==1.17.1 libcudnn9-cuda-12 & libcudnn9-dev-cuda-12
Cython==3.0.12 libgcc-11-dev:arm64 & libgcc-s1:arm64
filelock==3.17.0 libgcc-11-dev:arm64 & libgcc-s1:arm64
ninja==1.11.1.1 libstdc++-11-dev:arm64 & libstdc++6:arm64
numpy==2.2.3
opencv-python==4.11.0.86
packaging==24.2
pandas==2.2.3
pillow==11.1.0
requests==2.32.3
s3transfer==0.11.4
scipy==1.15.2
setuptools==75.8.2
torch==2.6.0+cu124
torch-model-archiver==0.12.0
torchaudio==2.6.0+cu124
torchserve==0.12.0
torchvision==0.21.0+cu124

v2.2-hf-tgi-3.0.1-pt-2.4.0-inf-gpu-py311

06 Mar 22:04
5b10ae6
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

These images are built to be used only on AWS SageMaker

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-tgi-inference:2.4.0-tgi3.0.1-gpu-py311-cu124-ubuntu22.04-v2.2
sha256:9f6007ca1658a46ed320985e7c8633890cda8d65fac12850a6b06d094ce7ce57

All tags:

2.4.0-tgi3.0.1-gpu-py311-cu124-ubuntu22.04-v2.2-2025-03-06-18-12-13
2.4-tgi3.0-gpu-py311-cu124-ubuntu22.04-v2
2.4-gpu-py311-cu124-ubuntu22.04-v2
2.4-tgi3.0-gpu-py311-cu124-ubuntu22.04
2.4.0-tgi3.0.1-gpu-py311-cu124-ubuntu22.04
2.4.0-tgi3.0.1-gpu-py311-cu124-ubuntu22.04-v2.2
2.4.0-gpu-py311-cu124-ubuntu22.04

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.2 libgcc-11-dev:amd64 & libgcc-s1:amd64
cffi==1.17.1 libgcc-11-dev:amd64 & libgcc-s1:amd64
cloudpickle==3.1.1 libstdc++-11-dev:amd64 & libstdc++6:amd64
datasets==2.21.0
filelock==3.16.1
numpy==1.26.4
packaging==24.1
pandas==2.2.3
pillow==11.0.0
protobuf==4.25.5
py-cpuinfo==9.0.0
regex==2024.9.11
requests==2.32.3
scipy==1.13.1
sentencepiece==0.2.0
setuptools==75.2.0
tokenizers==0.20.3
torch==2.4.0
transformers==4.46.3
wrapt==1.16.0

v1.9-pt-arm64-sagemaker-2.5.1-inf-cpu-py311

06 Mar 20:24
2b3fcc3
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-arm64:2.5.1-cpu-py311-ubuntu22.04-sagemaker-v1.9
sha256:f6e8314666c7e6f14cb6d4059e3fa2ac87747cd2bd171e1d7342c84870af685f

All tags:

2.5.1-cpu-py311-ubuntu22.04-sagemaker
2.5.1-cpu-py311-ubuntu22.04-sagemaker-v1.9-2025-03-06-19-33-30
2.5.1-cpu-py311-ubuntu22.04-sagemaker-v1.9
2.5-cpu-py311
2.5-cpu-py311-ubuntu22.04-sagemaker-v1
2.5.1-cpu-py311

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.36.7 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.35.66 libgcc-11-dev:arm64 & libgcc-s1:arm64
botocore==1.35.66 libgcc-11-dev:arm64 & libgcc-s1:arm64
cffi==1.17.1 libstdc++-11-dev:arm64 & libstdc++6:arm64
Cython==3.0.11
filelock==3.16.1
ninja==1.11.1.1
numpy==2.1.3
opencv-python==4.10.0.84
packaging==24.2
pandas==2.2.3
pillow==11.0.0
requests==2.32.3
s3transfer==0.10.4
scikit-learn==1.5.2
scipy==1.14.1
setuptools==75.6.0
torch==2.5.1+cpu
torch-model-archiver==0.12.0
torchaudio==2.5.1+cpu
torchserve==0.12.0
torchvision==0.20.1+cpu

v1.17-pt-graviton-sagemaker-2.4.0-inf-cpu-py311

06 Mar 21:52
355d7f9
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-graviton:2.4.0-cpu-py311-ubuntu22.04-sagemaker-v1.17
sha256:7ff727b65579520eefd6739f45b0ee2a1768b13238fbc10981b65479b20f7455

All tags:

2.4.0-cpu-py311-ubuntu22.04-sagemaker-v1.17-2025-03-06-21-03-08
2.4-cpu-py311
2.4.0-cpu-py311-ubuntu22.04-sagemaker
2.4-cpu-py311-ubuntu22.04-sagemaker-v1
2.4.0-cpu-py311
2.4.0-cpu-py311-ubuntu22.04-sagemaker-v1.17

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.35.13 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.35.47 libgcc-10-dev:arm64 & libgcc-11-dev:arm64 & libgcc-s1:arm64
botocore==1.35.47 libgcc-10-dev:arm64 & libgcc-11-dev:arm64 & libgcc-s1:arm64
cffi==1.17.1 libstdc++-10-dev:arm64 & libstdc++-11-dev:arm64 & libstdc++6:arm64
Cython==3.0.11
filelock==3.16.1
ninja==1.11.1.1
numpy==1.26.4
opencv-python==4.10.0.84
packaging==24.2
pandas==2.2.3
pillow==11.0.0
requests==2.32.3
s3transfer==0.10.3
scikit-learn==1.5.2
scipy==1.14.1
setuptools==75.2.0
torch==2.4.0+cpu
torch-model-archiver==0.12.0
torchaudio==2.4.0+cpu
torchserve==0.12.0
torchvision==0.19.0+cpu

v1.13-pt-sagemaker-2.4.0-inf-py311

06 Mar 23:29
355d7f9
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference:2.4.0-cpu-py311-ubuntu22.04-sagemaker-v1.13
sha256:84a6b0d11d7bdb5cfd1f6fd037ceed8f9a26d0c7e8448a57331a0eb56d559216

All tags:

2.4-cpu-py311-ubuntu22.04-sagemaker-v1
2.4.0-cpu-py311
2.4.0-cpu-py311-ubuntu22.04-sagemaker-v1.13-2025-03-06-21-15-14
2.4.0-cpu-py311-ubuntu22.04-sagemaker-v1.13
2.4.0-cpu-py311-ubuntu22.04-sagemaker
2.4-cpu-py311

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.35.12 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.35.46 libgcc-11-dev:amd64 & libgcc-s1:amd64
botocore==1.35.46 libgcc-11-dev:amd64 & libgcc-s1:amd64
cffi==1.17.1 libstdc++-11-dev:amd64 & libstdc++6:amd64
Cython==3.0.11
filelock==3.16.1
h5py==3.12.1
ninja==1.11.1.1
numpy==2.1.2
opencv-python==4.10.0.84
packaging==24.2
pandas==2.2.3
pillow==11.0.0
requests==2.32.3
s3transfer==0.10.3
scikit-learn==1.5.2
scipy==1.14.1
setuptools==75.2.0
torch==2.4.0+cpu
torch-model-archiver==0.12.0
torchaudio==2.4.0+cpu
torchserve==0.12.0
torchvision==0.19.0+cpu

  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference:2.4.0-gpu-py311-cu124-ubuntu22.04-sagemaker-v1.13
sha256:73b89cc528a67efadd2d71c78729d3415779c27e724dbac92b1337171409097c

All tags:

2.4.0-gpu-py311-cu124-ubuntu22.04-sagemaker
2.4.0-gpu-py311-cu124-ubuntu22.04-sagemaker-v1.13-2025-03-06-21-15-14
2.4.0-gpu-py311-cu124-ubuntu22.04-sagemaker-v1.13
2.4-gpu-py311
2.4.0-gpu-py311
2.4-gpu-py311-cu124-ubuntu22.04-sagemaker-v1

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 cuda-command-line-tools-12-4
awscli==1.35.12 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.35.46 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
botocore==1.35.46 libcublas-12-4 & libcublas-dev-12-4
cffi==1.17.1 libcudnn9-cuda-12 & libcudnn9-dev-cuda-12
Cython==3.0.11 libgcc-11-dev:amd64 & libgcc-s1:amd64
filelock==3.16.1 libgcc-11-dev:amd64 & libgcc-s1:amd64
h5py==3.12.1 libstdc++-11-dev:amd64 & libstdc++6:amd64
mpi4py==4.0.1
ninja==1.11.1.1
numpy==2.1.2
opencv-python==4.10.0.84
packaging==24.2
pandas==2.2.3
pillow==11.0.0
requests==2.32.3
s3transfer==0.10.3
scikit-learn==1.5.2
scipy==1.14.1
setuptools==75.2.0
torch==2.4.0+cu124
torch-model-archiver==0.12.0
torchaudio==2.4.0+cu124
torchserve==0.12.0
torchvision==0.19.0+cu124

v1.0-djl-0.32.0-inf-trt-0.12.0

06 Mar 23:26
a184c91
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/djl-inference:0.32.0-tensorrtllm0.12.0-cu125-v1.0
sha256:40efd86d1417b9ad2dbe8ce04f66196f58537acda4976746cb755ae449f27141

All tags:

0.32.0-tensorrtllm0.12.0-cu125-v1.0
0.32-tensorrtllm0.12.0-cu125-v1
0.32.0-tensorrtllm0.12.0-cu125
0.32.0-tensorrtllm0.12.0-cu125-v1.0-2025-03-06-19-49-32

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.2 cuda-command-line-tools-12-5
absl-py==2.1.0 libcublas-12-5 & libcublas-dev-12-5
awscli==1.38.6 libgcc-13-dev:amd64 & libgcc-s1:amd64
botocore==1.37.6 libgcc-13-dev:amd64 & libgcc-s1:amd64
cloudpickle==3.1.1 libstdc++-13-dev:amd64 & libstdc++6:amd64
datasets==2.19.1
filelock==3.17.0
h5py==3.13.0
mpi4py==4.0.3
ninja==1.11.1.3
numpy==1.26.4
packaging==24.2
pandas==2.2.3
pillow==11.1.0
protobuf==6.30.0
regex==2024.11.6
requests==2.32.3
s3transfer==0.11.4
scipy==1.15.2
sentencepiece==0.2.0
setuptools==75.8.2
tokenizers==0.19.1
torch==2.4.0
torchvision==0.19.0
transformers==4.44.2

v1.7-tf-ec2-2.18.0-inf-py310

05 Mar 21:08
d984c86
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference:2.18.0-cpu-py310-ubuntu20.04-ec2-v1.7
sha256:4136af2c0779c5df23e63ecc98711e7c284d0f4cf76027a682e1691b82f7c80d

All tags:

2.18.0-cpu-py310-ec2
2.18.0-cpu-py310-ubuntu20.04-ec2-v1.7-2025-03-05-20-23-09
2.18.0-cpu-ec2
2.18-cpu-py310-ubuntu20.04-ec2-v1
2.18-cpu-ec2
2.18.0-cpu-py310-ubuntu20.04-ec2
2.18-cpu-py310-ec2
2.18.0-cpu-py310-ubuntu20.04-ec2-v1.7

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.2 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.37.4 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.36.4 libgcc-9-dev:amd64 & libgcc-s1:amd64
botocore==1.36.4 libgcc-9-dev:amd64 & libgcc-s1:amd64
Cython==0.29.37 libstdc++-9-dev:amd64 & libstdc++6:amd64
packaging==24.2
protobuf==4.25.5
requests==2.32.3
s3transfer==0.11.1
setuptools==75.8.0
tensorflow-serving-api==2.18.0

  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference:2.18.0-gpu-py310-cu122-ubuntu20.04-ec2-v1.7
sha256:8abc9e6319672f03fc30cb4150ae2ba230ea0698b7a3fae1d196045ea8111126

All tags:

2.18.0-gpu-ec2
2.18-gpu-py310-ec2
2.18.0-gpu-py310-cu122-ubuntu20.04-ec2
2.18.0-gpu-py310-cu122-ubuntu20.04-ec2-v1.7
2.18-gpu-ec2
2.18.0-gpu-py310-ec2
2.18.0-gpu-py310-cu122-ubuntu20.04-ec2-v1.7-2025-03-05-20-23-16
2.18-gpu-py310-cu122-ubuntu20.04-ec2-v1

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.2 cuda-command-line-tools-12-2
awscli==1.37.4 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.36.4 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
botocore==1.36.4 libcublas-12-2 & libcublas-dev-12-2
Cython==0.29.37 libcudnn8 & libcudnn8-dev
packaging==24.2 libgcc-9-dev:amd64 & libgcc-s1:amd64
protobuf==4.25.5 libgcc-9-dev:amd64 & libgcc-s1:amd64
requests==2.32.3 libnccl-dev & libnccl2
s3transfer==0.11.1 libnccl-dev
setuptools==75.8.0 libstdc++-9-dev:amd64 & libstdc++6:amd64
tensorflow-serving-api-gpu==2.18.0

v1.4-tf-arm64-sagemaker-2.18.0-inf-cpu-py310

05 Mar 23:30
d984c86
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference-arm64:2.18.0-cpu-py310-ubuntu20.04-sagemaker-v1.4
sha256:973962a78942babedf55e5d84440b9272889009dd3c0654180de6a491b3916e6

All tags:

2.18-cpu
2.18.0-cpu-py310
2.18.0-cpu-py310-ubuntu20.04-sagemaker
2.18.0-cpu
2.18-cpu-py310-ubuntu20.04-sagemaker-v1
2.18.0-cpu-py310-ubuntu20.04-sagemaker-v1.4-2025-03-05-22-52-47
2.18.0-cpu-py310-ubuntu20.04-sagemaker-v1.4
2.18-cpu-py310

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.2 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.37.18 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.36.18 libgcc-9-dev:arm64 & libgcc-s1:arm64
botocore==1.36.18 libgcc-9-dev:arm64 & libgcc-s1:arm64
Cython==0.29.37 libstdc++-9-dev:arm64 & libstdc++6:arm64
packaging==24.2
protobuf==4.25.6
requests==2.32.3
s3transfer==0.11.2
setuptools==75.8.0
tensorflow-serving-api==2.18.0

v1.4-tf-arm64-ec2-2.18.0-inf-cpu-py310

05 Mar 23:19
d984c86
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-inference-arm64:2.18.0-cpu-py310-ubuntu20.04-ec2-v1.4
sha256:a5b4c422d83700fd22263156223a3fa9344497b60ff67eda4ad42b0e06287ee5

All tags:

2.18-cpu-ec2
2.18.0-cpu-py310-ubuntu20.04-ec2
2.18.0-cpu-ec2
2.18.0-cpu-py310-ubuntu20.04-ec2-v1.4-2025-03-05-22-43-12
2.18-cpu-py310-ubuntu20.04-ec2-v1
2.18-cpu-py310-ec2
2.18.0-cpu-py310-ubuntu20.04-ec2-v1.4
2.18.0-cpu-py310-ec2

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.2 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.37.18 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.36.18 libgcc-9-dev:arm64 & libgcc-s1:arm64
botocore==1.36.18 libgcc-9-dev:arm64 & libgcc-s1:arm64
Cython==0.29.37 libstdc++-9-dev:arm64 & libstdc++6:arm64
packaging==24.2
protobuf==4.25.6
requests==2.32.3
s3transfer==0.11.2
setuptools==75.8.0
tensorflow-serving-api==2.18.0

v1.34-pt-sagemaker-2.3.0-inf-py311

05 Mar 20:29
d984c86
Compare
Choose a tag to compare

For details on the complete list of DLCs, please visit the AWS documentation page: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html

Images Published:


  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference:2.3.0-cpu-py311-ubuntu20.04-sagemaker-v1.34
sha256:a8b50a4a43e03ef12eedb5ff365fb552dbaa652afbf0447401cbbc4ffa4c79a1

All tags:

2.3.0-cpu-py311
2.3-cpu-py311-ubuntu20.04-sagemaker-v1
2.3-cpu-py311
2.3.0-cpu-py311-ubuntu20.04-sagemaker-v1.34-2025-03-05-18-13-34
2.3.0-cpu-py311-ubuntu20.04-sagemaker-v1.34
2.3.0-cpu-py311-ubuntu20.04-sagemaker

Important Packages:

pip packages apt|deb packages
PyYAML==6.0 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.33.4 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.34.122 libgcc-9-dev:amd64 & libgcc-s1:amd64
botocore==1.34.122 libgcc-9-dev:amd64 & libgcc-s1:amd64
cffi==1.16.0 libstdc++-9-dev:amd64 & libstdc++6:amd64
Cython==3.0.10
filelock==3.14.0
h5py==3.11.0
ninja==1.11.1.1
numpy==1.26.4
opencv-python==4.10.0.82
packaging==24.2
pandas==2.2.2
pillow==10.3.0
requests==2.32.3
s3transfer==0.10.1
sagemaker-pytorch-inference==2.0.24
scikit-learn==1.5.0
scipy==1.13.1
setuptools==75.8.2
torch==2.3.0+cpu
torch-model-archiver==0.11.0
torchaudio==2.3.0+cpu
torchserve==0.11.0
torchvision==0.18.0+cpu

  1. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference:2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker-v1.34
sha256:6c998b151732ad78b48d7506411db2db86d3599a4d19ef2c90535770a907ff01

All tags:

2.3.0-gpu-py311
2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker-v1.34-2025-03-05-18-13-27
2.3-gpu-py311
2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker-v1.34
2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker
2.3-gpu-py311-cu121-ubuntu20.04-sagemaker-v1

Important Packages:

pip packages apt|deb packages
PyYAML==6.0 cuda-command-line-tools-12-1
awscli==1.33.4 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
boto3==1.34.122 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
botocore==1.34.122 libcublas-12-1 & libcublas-dev-12-1
cffi==1.16.0 libcudnn8
Cython==3.0.10 libgcc-9-dev:amd64 & libgcc-s1:amd64
filelock==3.14.0 libgcc-9-dev:amd64 & libgcc-s1:amd64
h5py==3.11.0 libstdc++-9-dev:amd64 & libstdc++6:amd64
ninja==1.11.1.1
numpy==1.26.4
opencv-python==4.10.0.82
packaging==24.2
pandas==2.2.2
pillow==10.3.0
requests==2.32.3
s3transfer==0.10.1
sagemaker-pytorch-inference==2.0.24
scikit-learn==1.5.0
scipy==1.13.1
setuptools==75.8.2
torch==2.3.0+cu121
torch-model-archiver==0.11.0
torchaudio==2.3.0+cu121
torchserve==0.11.0
torchvision==0.18.0+cu121