AMI de aprendizaje profundo de AWS - GPU TensorFlow 2.16 (Ubuntu 20.04) - AWS Deep Learning AMIs
Formato del nombre de AMIInstancias de EC2 admitidasContenido de AMIAvisoFecha de la versión: 17/02/2025Fecha de la versión: 20/01/2025Fecha de la versión: 22/10/2024Fecha de la versión: 04/10/2024Fecha de la versión: 20/09/2024Fecha de la versión: 19/08/2024Versión 2.16.2. - Fecha de lanzamiento: 25/07/2024Versión 2.16.1 - Fecha de lanzamiento: 06/06/2024Fecha de la versión: 2024-05-10

AMI de aprendizaje profundo de AWS - GPU TensorFlow 2.16 (Ubuntu 20.04)

Para obtener ayuda para comenzar, consulte Introducción a las DLAMI.

Formato del nombre de AMI

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) ${YYYY-MM-DD}

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) ${YYYY-MM-DD}

Instancias de EC2 admitidas

  • Consulte el documento de cambios importantes en la DLAMI.

  • El aprendizaje profundo con el controlador con software de código abierto de Nvidia admite G4dn, G5, G6, Gr6, G6e, P4d, P4de, P5, P5e y P5en.

  • El aprendizaje profundo con el controlador propietario de Nvidia admite G3 (no admite G3.16x), P3 y P3dn.

La AMI incluye lo siguiente:

  • Servicio de AWS compatible: EC2

  • Sistema operativo: Ubuntu 20.04

  • Arquitectura de cómputo: x86

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

  • Versión de TensorFlow: 2.16

  • Controlador de NVIDIA:

    • Controlador con software de código abierto de Nvidia: 550.144.03

    • Controlador propietario Nvidia: 550.144.03

  • Pila CUDA12 de NVIDIA:

    • Ruta de instalación de CUDA, NCCL y cuDDN: /usr/local/cuda-12.3/

  • Instalador de EFA: 1.34.0

  • AWS CLI v2 como aws2 y AWS CLI v1 como aws

  • Tipo de volumen EBS: gp3

  • AMI-ID de la consulta con el parámetro SSM (la región de ejemplo es us-east-1):

    • Controlador con software de código abierto de Nvidia:

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

    • Controlador propietario Nvidia:

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

  • AMI-ID de la consulta con AWSCLI (la región de ejemplo es us-east-1):

    • Controlador con software de código abierto de Nvidia:

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

    • Controlador propietario Nvidia:

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

Aviso

La versión de Keras está anclada a la 2.0 en lugar de a la 3.0

Con la última versión de TF2.16, Keras se ha actualizado de la versión principal 2 a la versión principal 3.0. Esta versión de Keras es una reescritura completa del paquete de Keras (consulte la documentación de Keras 3 para obtener más información). Para garantizar la compatibilidad con los flujos de trabajo de los clientes, hemos anclado las versiones de Keras en la 2.0 mediante la variable de entorno TF_USE_LEGACY_KERAS=1. Si sus flujos de trabajo requieren usar Keras 3.0, elimine esta variable de entorno de su entorno virtual de TensorFlow /opt/tensorflow mediante el siguiente script:

source /opt/tensorflow/bin/activate
unset TF_USE_LEGACY_KERAS

Fecha de la versión: 17/02/2025

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20250215

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20250215

Actualizado
Eliminaciones

Fecha de la versión: 20/01/2025

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20250118

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20250118

Actualizado

Fecha de la versión: 22/10/2024

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20241022

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20241022

Actualizado

Fecha de la versión: 04/10/2024

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20241004

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240920

Actualizado

  • Se ha actualizado el kit de herramientas de contenedores de Nvidia de la versión 1.16.1 a la 1.16.2 para corregir la vulnerabilidad de seguridad CVE-2024-0133.

Fecha de la versión: 20/09/2024

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240920

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240920

Actualizado

  • Se han actualizado el controlador Nvidia y Fabric Manager de la versión 535.183.01 a la 550.90.07.

  • Se ha actualizado EFA de la versión 1.32.0 a la 1.34.0.

  • Se ha actualizado PyTorch de la versión 2.3.0 a la 2.3.1.

Se agregó

  • Se ha añadido compatibilidad para la instancia EC2 P5e en las imágenes del controlador con software de código abierto de Nvidia.

Fecha de la versión: 19/08/2024

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240816

Se agregó

Versión 2.16.2. - Fecha de lanzamiento: 25/07/2024

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240725

Actualizado

  • Se ha actualizado la versión del parche TensorFlow de la versión 2.16.1 a la 2.16.2

  • Se ha resuelto que la versión secundaria de TensorFlow se actualice de 2.16 a 2.17

    • La versión de la AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240717 contenía inadvertidamente la versión secundaria 2.17 de TensorFlow en lugar de la 2.16. Asegúrese de que los flujos de trabajo que dependen de TensorFlow 2.16 se actualicen a la última DLAMI.

Versión 2.16.1 - Fecha de lanzamiento: 06/06/2024

Nombres de AMI:

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240606

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) 20240606

Actualizado

  • Se ha actualizado la versión del controlador Nvidia de la 535.183.01 a la 535.161.08.

Fecha de la versión: 2024-05-10

Nombres de AMI:

Consulte el documento de cambios importantes en la DLAMI.

  • AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) <>

  • AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04) <>

Se agregó

  • Publicación inicial de:

    • Serie de AMI de aprendizaje profundo del controlador propietario de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04).

    • Serie de AMI de aprendizaje profundo del controlador con software de código abierto de Nvidia - GPU TensorFlow 2.16 (Ubuntu 20.04).

    • El software incluye lo siguiente:

      • “nvidia-driver=535.161.08”

      • "fabric-manager=535.161.08”

      • “cuda=12.3”

      • “cudnn=8.9.7”

      • “efa=1.32.0”

      • “nccl=2.21.5”

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

  • Se ha añadido el entorno virtual tensorflow (fuente del comando de activación /opt/tensorflow/bin/activate). Este entorno incluye lo siguiente:

    • “tensorflow=2.16.1”

    • NOTA

      • A partir de TF2.16, se elimina la API tf.estimator.

      • Para garantizar la compatibilidad con los flujos de trabajo de los clientes, hemos anclado las versiones de Keras en la 2.0 mediante la variable de entorno TF_USE_LEGACY_KERAS=1. Si sus flujos de trabajo requieren usar Keras 3.0, elimine esta variable de entorno de su entorno virtual de TensorFlow /opt/tensorflow mediante el siguiente script:

source /opt/tensorflow/bin/activate
unset TF_USE_LEGACY_KERAS