深度学习 AMI GPU TensorFlow 2.18 (Ubuntu 22.04) - Amazon Deep Learning AMIs
AMI 名称格式支持的 EC2 实例AMI 内容发布日期：2025-02-17发布日期：2025-01-20发布日期：2024-12-09
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异，请参阅 中国的 Amazon Web Services 服务入门 (PDF)。

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

深度学习 AMI GPU TensorFlow 2.18 (Ubuntu 22.04)

如需入门帮助，请参阅DLAMI 入门。

AMI 名称格式

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

支持的 EC2 实例

  • 使用 OSS 进行深度学习 Nvidia Driver 支持 G4dn、G5、G6、Gr6、G6e、p4d、p4de、P5、p5e、p5e、p5eN。

AMI 包括以下内容：

  • 支持的 Amazon 服务：Amazon EC2

  • 操作系统：Ubuntu 22.04

  • 计算架构：x86

  • Python：/opt/tensorflow/bin/python3.12

  • TensorFlow 版本：2.18

  • 英伟达驱动程序：

    • OSS Nvidia 驱动程序：570.172.08

  • 英伟达 CUDA12 堆栈：

    • CUDA、NCCL 和 cudDN 安装路径：/-12.5/ usr/local/cuda

  • EFA 安装程序：1.43.1

  • Amazon CLI v2 是 aws2，v1 是 aws Amazon CLI

  • EBS 卷类型：gp3

  • 使用 SSM 参数查询 AMI-ID（示例区域为 us-east-1）：

    • OSS 英伟达驱动程序：

      aws ssm get-parameter --region us-east-1 \
    --name /aws/service/deeplearning/ami/x86_64/oss-nvidia-driver-gpu-tensorflow-2.18-ubuntu-22.04/latest/ami-id \
    --query "Parameter.Value" \
    --output text

  • 使用以下方式查询 AMI-ID AWSCLI （示例区域为 us-east-1）：

    • OSS 英伟达驱动程序：

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

发布日期：2025-02-17

AMI 名称：深度学习 OSS Nvidia 驱动程序 AMI GPU TensorFlow 2.18 (Ubuntu 22.04) 20250215

已更新

  • 将 NVIDIA 容器工具包从版本 1.17.3 更新到 1.17.4 版

已删除

发布日期：2025-01-20

AMI 名称：深度学习 OSS Nvidia 驱动程序 AMI GPU TensorFlow 2.18 (Ubuntu 22.04) 20250118

已更新

  • 将 Nvidia 驱动程序从 550.90.07 版本升级到 550.127.05，以解决 2025 年 1 月 NVIDIA G PU CVEs 显示驱动程序安全公告中提到的问题

发布日期：2024-12-09

AMI 名称：深度学习 OSS Nvidia 驱动程序 AMI GPU TensorFlow 2.18 (Ubuntu 22.04) 20241206

新增了

  • 深度学习 OSS Nvidia Driver AMI GPU TensorFlow 2.18 (Ubuntu 22.04) 系列的首次发布。

    • 软件包括以下内容：

      • “nvidia-driver=550.127.05”

      • “fabric-manager=550.127.05”

      • “cuda=12.5"

      • “cudnn=9.5.1"

      • “efa=1.37.0"

      • “nccl=2.23.4"

      • “aws-nccl-ofi-plugin=v1.13.0-aws”

  • Tensorflow 虚拟环境（激活命令源/opt/tensorflow/bin/activate）包括以下内容：

    • “tensorflow=2.18.0”

Fixed

  • 由于 Ubuntu 内核为解决内核地址空间布局随机化 (KASLR) 功能中的缺陷而进行了更改，G4Dn/G5 实例无法在 OSS Nvidia 驱动程序上正确初始化 CUDA。为了缓解此问题，此 DLAMI 包含动态加载 G4Dn 和 G5 实例专有驱动程序的功能。请留出一段短暂的初始化时间进行此加载，以确保您的实例能够正常运行。

    • 要检查此服务的状态和运行状况，可以使用以下命令：

sudo systemctl is-active dynamic_driver_load.service
active