View a markdown version of this page

AWS Deep Learning OSS AMI GPU PyTorch 2.14 (Amazon Linux 2023) - AWS Deep Learning AMIs
NoticesAMI Name FormatSSM Parameter Query AWS CLI Parameter QueryLatest releases

AWS Deep Learning OSS AMI GPU PyTorch 2.14 (Amazon Linux 2023)

This Deep Learning AMI provides a GPU-enabled PyTorch 2.14 environment on Amazon Linux 2023.

Note

For detailed software package information, see the individual release notes pages that follow.

For help getting started, see Getting started with DLAMI.

Notices

  • CUDA is no longer available at the system level on this AMI, so tools that expect it there will not find it. Starting with PyTorch 2.9, CUDA and related libraries live in the PyTorch virtual environment at /opt/pytorch/cuda. This change reduces the image size while maintaining full CUDA functionality. To use CUDA, activate the PyTorch virtual environment using source /opt/pytorch/bin/activate.

AMI Name Format

  • Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.14 (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.14-amazon-linux-2023/latest/ami-id && \
        aws ssm get-parameter --region us-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 --region us-east-1 --owners amazon --filters 'Name=name,Values=Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.14 (Amazon Linux 2023) ????????' 'Name=state,Values=available' --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' --output text
Latest releases
Latest Release Notes for this DLAMI