View a markdown version of this page

AWS 深度学习 OSS AMI GPU PyTorch 2.13 (亚马逊 Linux 2023) - AWS Deep Learning AMIs

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

AWS 深度学习 OSS AMI GPU PyTorch 2.13 (亚马逊 Linux 2023)

注意

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

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

通知
  • 从 PyTorch 2.9 开始CUDA,相关库位于PyTorch虚拟环境中,/opt/pytorch/cuda而不是系统级别。此更改减小了图像大小,同时保持了全部CUDA功能。要使用CUDA,请使用激活PyTorch虚拟环境source /opt/pytorch/bin/activate

AMI 名称格式
  • 深度学习 OSS Nvidia 驱动程序 AMI GPU PyTorch 2.13(亚马逊 Linux 2023)$ {} YYYY-MM-DD

SSM 参数查询

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

export SSM_PARAMETER=oss-nvidia-driver-gpu-pytorch-2.13-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 查询

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

aws ec2 describe-images --region us-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
最新版本