AWS 深度学习 ARM64 AMI OSS GPU PyTorch 2.7 (Ubuntu 22.04) - AWS Deep Learning AMIs

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

AWS 深度学习 ARM64 AMI OSS GPU PyTorch 2.7 (Ubuntu 22.04)

如需入门帮助,请参阅DLAMI 入门

AMI 名称格式

  • 深度学习 ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.7 (Ubuntu 22.04) $ {YYYY-MM-DD}

支持的 EC2 实例

  • G5g

AMI 包括以下内容:

  • 支持的 AWS 服务:Amazon EC2

  • 操作系统:Ubuntu 22.04

  • 计算架构: ARM64

  • Linux 内核:6.8

  • NVIDIA 驱动程序:570.133.20

  • NVIDIA CUDA 12.8 堆栈

    • C@@ UDA、NCCL 和 cudDN 安装目录:/-12.8/ usr/local/cuda

    • 默认 CUDA:12.8

      • 路径/ usr/local/cuda 指向 CUDA 12.8

      • 在环境变量下方更新:

        • LD_LIBRARY_PATH 要有/-usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda:/usr/local/cuda/targets/sbsa

        • 要拥有的路径/usr/local/cuda/bin/:/usr/local/cuda/include/

  • AWS CLI v2 位于/usr/local/bin/aws

  • EBS 卷类型:gp3

  • Nvidia 容器工具包:1.17.7

    • 版本命令: nvidia-container-cli-V

  • Docker:28.2.2

  • Python:/usr/bin/python3.12

  • 使用 SSM 参数查询 AMI-ID(示例区域为 us-east-1):

    aws ssm get-parameter --region us-east-1 \ --name /aws/service/deeplearning/ami/arm64/oss-nvidia-driver-gpu-pytorch-2.7-ubuntu-22.04/latest/ami-id \ --query "Parameter.Value" \ --output text
  • 使用以下方式查询 AMI-ID AWSCLI (示例区域为 us-east-1):

    aws ec2 describe-images --region us-east-1 --owners amazon --filters 'Name=name,Values=Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.7 (Ubuntu 22.04) ????????' 'Name=state,Values=available' --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' --output text

版权声明

PyTorch 弃用 Anaconda 频道

从 PyTorch 2.6 开始, PyTorch 已弃用对 Conda 的支持(参见官方公告)。因此, PyTorch 2.6 及更高版本将转向使用 Python 虚拟环境。要激活 PyTorch venv,请使用源代码/ opt/pytorch/bin/activate

发布日期:2025-05-22

AMI 名称:深度学习 ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.7 (Ubuntu 22.04) 20250521

新增了

  • 深度学习 ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.7(Ubuntu 22.04)系列的首次发布。包括配备 NVIDIA 驱动程序 R570、CUDA=12.8、cudnn=9.10 和 NCCL=2.26.2 的 Python 虚拟环境 pytorch(来源/opt/pytorch/bin/activate) PyTorch