Skip to content

v1.0-trcomp-hf-4.21.1-pt-1.11.0-tr-gpu-py38

Compare
Choose a tag to compare
@aws-dlinfra-bot aws-dlinfra-bot released this 30 Aug 00:10
1e22d3b

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-trcomp-training:1.11.0-transformers4.21.1-gpu-py38-cu113-ubuntu20.04-v1.0
sha256:638b5a6e76b95a46a5291815fed695e99f85e1097ee1ef151bca465dc0969083

All tags:

1.11-gpu-py38
1.11.0-transformers4.21.1-gpu-py38-cu113-ubuntu20.04-v1.0
1.11.0-gpu-py38-cu113-ubuntu20.04
1.11.0-transformers4.21.1-gpu-py38-cu113-ubuntu20.04
1.11-transformers4.21-gpu-py38-cu113-ubuntu20.04-v1
1.11.0-gpu-py38
1.11-gpu-py38-cu113-ubuntu20.04-v1
1.11-transformers4.21-gpu-py38-cu113-ubuntu20.04
1.11.0-transformers4.21.1-gpu-py38-cu113-ubuntu20.04-v1.0-2022-08-29-20-42-01

Important Packages:

pip packages apt|deb packages
PyYAML==5.4.1 cuda-command-line-tools-11-3
absl-py==1.2.0 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
apex==0.1 emacs & emacs-bin-common & emacs-common & emacs-gtk & emacsen-common
awscli==1.25.56 libcublas-11-3 & libcublas-dev-11-3
awsio==0.0.1 libcudnn8
boto3==1.24.55 libgcc-9-dev:amd64 & libgcc-s1:amd64
botocore==1.27.55 libgcc-9-dev:amd64 & libgcc-s1:amd64
cffi==1.15.1 libstdc++-9-dev:amd64 & libstdc++6:amd64
cloudpickle==2.1.0
Cython==0.29.32
datasets==1.18.4
dgl==0.8.0.post2
filelock==3.8.0
h5py==3.7.0
horovod==0.24.3
mpi4py==3.1.3
numpy==1.22.2
opencv-python==4.6.0.66
packaging==21.3
pandas==1.4.3
Pillow==9.2.0
protobuf==3.19.4
pybind11==2.10.0
regex==2022.8.17
requests==2.28.1
s3fs==0.4.2
s3transfer==0.6.0
sagemaker==2.106.0
sagemaker-experiments==0.1.35
sagemaker-pytorch-training==2.6.2
sagemaker-training==4.2.6
scikit-learn==1.1.2
scipy==1.9.0
seaborn==0.11.2
sentencepiece==0.1.97
setuptools==65.0.2
smclarify==0.2
smdebug==1.0.16b20220819
smdebug-rulesconfig==1.0.1
smdistributed-dataparallel==1.4.1
smdistributed-modelparallel==1.10.1
tokenizers==0.12.1
torch==1.11.0+cu113
torchaudio==0.11.0+6297e97
torchnet==0.0.4
torchvision==0.12.0a0+9b5a3fe
transformers==4.21.1