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

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

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

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

AMI 名称格式

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

支持的 EC2 实例

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

AMI 包括以下内容:

  • 支持的 Amazon 服务:Amazon EC2

  • 操作系统:Ubuntu 22.04

  • 计算架构:x86

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

  • TensorFlow 版本:2.17

  • 英伟达驱动程序: 

    • OSS Nvidia 驱动程序:570.172.08

  • 英伟达 CUDA12 堆栈

    • CUDA、NCCL 和 cudDN 安装路径:/-12.3/ usr/local/cuda

  • EFA 安装程序:1.34.0

  • 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.17-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.17 (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.17 (Ubuntu 22.04) 20250215

已更新

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

已删除

发布日期:2025-01-20

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

已更新

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

版本 2.17.1

发布日期:2024-11-18

AMI 名称:深度学习 OSS Nvidia 驱动程序 AMI GPU TensorFlow 2.17 (Ubuntu 22.04) 20241115

Fixed 

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

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

sudo systemctl is-active dynamic_driver_load.service active

版本 2.17.0

发布日期:2024-09-25

AMI 名称:深度学习 OSS Nvidia 驱动程序 AMI GPU TensorFlow 2.17 (Ubuntu 22.04) 20240924

新增了

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

    • 软件包括以下内容:

      • “nvidia-driver=550.90.07”

      • “fabric-manager=550.90.07”

      • “cuda=12.3"

      • “cudnn=8.9.7"

      • “efa=1.34.0"

      • “nccl=2.22.3"

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

  • Tensorflow 虚拟环境(激活命令源/opt/tensorflow/bin/activate)包括以下内容:

    • “tensorflow=2.17.0”