Skip to content

v2.0-hf-4.55.4-pt-2.8.0-tr-neuronx-sdk2.26.0-py310

Latest

Choose a tag to compare

@aws-dlinfra-bot aws-dlinfra-bot released this 07 Nov 05:32
31952b2

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-training-neuronx:2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04-v2.0
    Image Digest:
sha256:935759c17b79afedf9fc3b8393b996290b3cb8e5be8495d662672745b75f6623

All tags:

2.8-transformers4.55-neuronx-py310-sdk2.26.0-ubuntu22.04
2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04
2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04-v2.0-2025-11-07-02-22-12
2.8-transformers4.55-neuronx-py310-sdk2.26.0-ubuntu22.04-v2
2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04-v2.0
2.8-neuronx-py310-sdk2.26.0-ubuntu22.04-v2
2.8.0-neuronx-py310-sdk2.26.0-ubuntu22.04

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.3 emacs-common & emacsen-common
absl-py==2.3.1 emacs-common & emacsen-common
awscli==1.36.34 libgcc-11-dev:amd64 & libgcc-s1:amd64
boto3==1.35.93 libgcc-11-dev:amd64 & libgcc-s1:amd64
botocore==1.35.93 libnccl-ofi
cffi==2.0.0 libstdc++-11-dev:amd64 & libstdc++6:amd64
cloudpickle==3.1.2
Cython==3.2.0
datasets==4.1.1
filelock==3.20.0
numpy==1.26.4
opencv-python==4.11.0.86
packaging==25.0
pandas==2.3.3
pillow==10.3.0
protobuf==3.20.3
regex==2025.11.3
requests==2.32.5
s3transfer==0.10.4
sagemaker==2.250.0
scikit-learn==1.7.2
scipy==1.12.0
seaborn==0.13.2
sentencepiece==0.2.1
setuptools==80.9.0
smdebug-rulesconfig==1.0.1
tokenizers==0.21.4
torch==2.8.0
torchvision==0.23.0
transformers==4.55.4