View a markdown version of this page

Amazon 深度学习 OSS AMI GPU PyTorch 2.14(亚马逊 Linux 2023) - Amazon Deep Learning AMIs
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)。

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

Amazon 深度学习 OSS AMI GPU PyTorch 2.14(亚马逊 Linux 2023)

这款深度学习 AMI 在亚马逊 Linux 2023 上提供了 GPU-enabled PyTorch2.14 环境。

注意

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

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

通知
  • CUDA已在此 AMI 的系统级别上不再可用,因此期望它的工具将找不到它。从 PyTorch 2.9 开始CUDA,相关库位于PyTorch虚拟环境中/opt/pytorch/cuda。此更改在保持完整CUDA功能的同时减小了图像大小。要使用CUDA,请使用激活PyTorch虚拟环境source /opt/pytorch/bin/activate。

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

SSM 参数查询

使用以下公共 SSM 参数查询来查找最新的 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
Amazon CLI 查询

使用以下 Amazon 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.14 (Amazon Linux 2023) ????????' 'Name=state,Values=available' --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' --output text
最新版本