AWS Deep Learning OSS AMI GPU PyTorch 2.13 (Amazon Linux 2023)
Note
For detailed software package information, see the individual release notes pages linked below.
For help getting started, see Getting started with DLAMI.
Notices
-
Starting with PyTorch 2.9, CUDA and related libraries are located in the PyTorch virtual environment at
/opt/pytorch/cudainstead of at the system level. This change reduces the image size while maintaining full CUDA functionality. To use CUDA, activate the PyTorch virtual environment usingsource /opt/pytorch/bin/activate.
AMI Name Format
-
Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.13 (Amazon Linux 2023) ${YYYY-MM-DD}
SSM Parameter Query
Use the following Public SSM Parameter Query to find the latest DLAMI:
export SSM_PARAMETER=oss-nvidia-driver-gpu-pytorch-2.13-amazon-linux-2023/latest/ami-id && \ aws ssm get-parameter --regionus-east-1\ --name /aws/service/deeplearning/ami/x86_64/$SSM_PARAMETER \ --query "Parameter.Value" \ --output text
AWS CLI query
Use the following AWS CLI query to find the latest DLAMI:
aws ec2 describe-images --regionus-east-1--owners amazon --filters 'Name=name,Values=Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.13 (Amazon Linux 2023) ????????' 'Name=state,Values=available' --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' --output text