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:
763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-training-neuronx:2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04-v1.0
sha256:26818c7930a615eda5c69446fb0db848c94dd26dd0f98e37c193a905cdee8cd5
All tags:
2.1.2-neuronx-py310-sdk2.20.0-ubuntu20.04
2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04
2.1-neuronx-py310-sdk2.20.0-ubuntu20.04-v1
2.1-transformers4.43-neuronx-py310-sdk2.20.0-ubuntu20.04
2.1-transformers4.43-neuronx-py310-sdk2.20.0-ubuntu20.04-v1
2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04-v1.0-2024-11-22-22-07-19
2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04-v1.0
Important Packages:
| pip packages | apt|deb packages |
|---|---|
| PyYAML==6.0.2 | emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common |
| absl-py==2.1.0 | emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common |
| awscli==1.34.19 | libgcc-8-dev:amd64 & libgcc-9-dev:amd64 & libgcc-s1:amd64 |
| boto3==1.35.19 | libgcc-8-dev:amd64 & libgcc-9-dev:amd64 & libgcc-s1:amd64 |
| botocore==1.35.19 | libstdc++-9-dev:amd64 & libstdc++6:amd64 |
| cffi==1.17.1 | |
| cloudpickle==2.2.1 | |
| datasets==2.18.0 | |
| filelock==3.16.0 | |
| h5py==3.11.0 | |
| numpy==1.25.2 | |
| opencv-python==4.10.0.84 | |
| packaging==24.1 | |
| pandas==2.2.2 | |
| pillow==10.3.0 | |
| protobuf==3.20.3 | |
| regex==2024.9.11 | |
| requests==2.31.0 | |
| s3transfer==0.10.2 | |
| sagemaker==2.232.2 | |
| scikit-learn==1.5.2 | |
| scipy==1.11.2 | |
| seaborn==0.13.2 | |
| sentencepiece==0.2.0 | |
| setuptools==69.5.1 | |
| smdebug-rulesconfig==1.0.1 | |
| tokenizers==0.19.1 | |
| torch==2.1.2 | |
| torchvision==0.16.2 | |
| transformers==4.43.2 |