本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
Amazon 深度学习 OSS AMI GPU PyTorch 2.14(亚马逊 Linux 2023)
这款深度学习 AMI 在亚马逊 Linux 2023 上提供了 GPU-enabled PyTorch
注意
有关软件包的详细信息,请参阅随后的各个发行说明页面。
有关入门帮助,请参阅 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 --regionus-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 --regionus-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