AWS 深度学习 OSS AMI ARM64 AMI GPU Pytorch 2.9(亚马逊 Linux 2023) - AWS Deep Learning AMIs

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

AWS 深度学习 OSS AMI ARM64 AMI GPU Pytorch 2.9(亚马逊 Linux 2023)

注意

有关软件包的详细信息,请参阅下面链接的各个发布说明页面

有关入门帮助,请参阅 DLAMI 入门

通知
  • 从 PyTorch 2.9 开始,CUDA 和相关库现在位于中的 Pytorch 虚拟环境中,/opt/pytorch/cuda而不是安装在系统级别。此更改减小了图像大小,同时保持了完整的 CUDA 功能。请确保使用激活 PyTorch 虚拟环境以使用 CUDA source /opt/pytorch/bin/activate

AMI 名称格式
  • 深度学习 ARM64 OSS Nvidia Driver AMI GPU PyTorch 2.9(亚马逊 Linux 2023)$ {YYYY-MM-DD}

SSM 参数查询

请使用以下公共 SSM 参数查询来查找最新的 DLAMI:

export SSM_PARAMETER=oss-nvidia-driver-gpu-pytorch-2.9-amazon-linux-2023/latest/ami-id && \ aws ssm get-parameter --region us-east-1 \ --name /aws/service/deeplearning/ami/arm64/$SSM_PARAMETER \ --query "Parameter.Value" \ --output text
AWSCLI 查询

请使用以下 AWS 查询来查找最新的 DLAMI:

aws ec2 describe-images --region us-east-1 --owners amazon --filters 'Name=name,Values=Deep Learning OSS Nvidia Driver AMI ARM64 AMI GPU Pytorch 2.9 (Amazon Linux 2023) ????????' 'Name=state,Values=available' --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' --output text
最新发布