本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。
AWS 深度學習 OSS AMI GPU PyTorch 2.13 (Amazon 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 (Amazon 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 --regionus-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 --regionus-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