Skip to content

v1.4-djl-0.27.0-inf-ds-0.12.6

Compare
Choose a tag to compare
@aws-dlinfra-bot aws-dlinfra-bot released this 09 May 22:10
51178a7

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.27.0-deepspeed0.12.6-cu121-v1.4
sha256:fb806b154e0aa518520802c55d213f9054a7aaf309153c29b4ed1607c5a058c3

All tags:

0.27.0-deepspeed0.12.6-cu121
0.27-deepspeed0.12.6-cu121-v1
0.27.0-deepspeed0.12.6-cu121-v1.4
0.27.0-deepspeed0.12.6-cu121-v1.4-2024-05-09-19-58-49

Important Packages:

pip packages apt|deb packages
PyYAML==6.0.1 cuda-command-line-tools-12-1
awscli==1.32.101 libcublas-12-1 & libcublas-dev-12-1
botocore==1.34.101 libcudnn8 & libcudnn8-dev
cloudpickle==3.0.0 libgcc-11-dev:amd64 & libgcc-s1:amd64
datasets==2.17.1 libgcc-11-dev:amd64 & libgcc-s1:amd64
filelock==3.14.0 libnccl-dev & libnccl2
mpi4py==3.1.6 libnccl-dev
ninja==1.11.1.1 libstdc++-11-dev:amd64 & libstdc++6:amd64
numpy==1.26.4
packaging==24.0
pandas==2.2.2
pillow==10.3.0
protobuf==3.20.3
py-cpuinfo==9.0.0
regex==2024.4.28
requests==2.31.0
s3transfer==0.10.1
scipy==1.13.0
sentencepiece==0.2.0
setuptools==69.5.1
tokenizers==0.15.2
torch==2.1.2+cu121
torchvision==0.16.2+cu121
transformers==4.39.0