View a markdown version of this page

AWS GPU AMI de aprendizaje profundo TensorFlow 2.16 (Amazon Linux 2) - AWS Deep Learning AMIs

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

AWS GPU AMI de aprendizaje profundo TensorFlow 2.16 (Amazon Linux 2)

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

Formato de nombre AMI

  • Controlador Nvidia propietario de aprendizaje profundo AMI GPU TensorFlow 2.16 (Amazon Linux 2) $ {} YYYY-MM-DD

  • Controlador Nvidia AMI GPU TensorFlow 2.16 para OSS de aprendizaje profundo (Amazon Linux 2) $ {} YYYY-MM-DD

Instancias de EC2 admitidas

  • Consulte Cambios importantes en las DLAMI.

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

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

La AMI incluye lo siguiente:

  • Servicio compatible: EC2 AWS

  • Sistema operativo: Amazon Linux 2

  • Arquitectura de computación: x86

  • Python:/opt/tensorflow/1.0 bin/python3

  • TensorFlow versión: 2.16

  • Controlador de NVIDIA:

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

    • Controlador NVIDIA propietario: 550.144.03

  • Pila CUDA12 de NVIDIA:

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

  • Instalador de EFA: 1.34.0

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

  • Tipo de volumen EBS: gp3

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

    • Controlador NVIDIA de código abierto:

      aws ssm get-parameter --name /aws/service/deeplearning/ami/x86_64/oss-nvidia-driver-gpu-tensorflow-2.16-amazon-linux-2/latest/ami-id --region us-east-1 --query "Parameter.Value" --output text
    • Controlador NVIDIA propietario:

      aws ssm get-parameter --name /aws/service/deeplearning/ami/x86_64/proprietary-nvidia-driver-gpu-tensorflow-2.16-amazon-linux-2/latest/ami-id --region us-east-1 --query "Parameter.Value" --output text
  • Consulta AMI-ID con AWSCLI (la región de ejemplo es us-east-1):

    • Controlador NVIDIA de código abierto:

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

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

Aviso

Kit de herramientas de contenedores de NVIDIA: 1.17.4

En la versión 1.17.4 del kit de herramientas de contenedores, el montaje de bibliotecas de compatibilidad con CUDA se ha desactivado. Para garantizar la compatibilidad con varias versiones de CUDA en los flujos de trabajo de contenedores, asegúrese de actualizar su LD_LIBRARY_PATH para incluir sus bibliotecas de compatibilidad con CUDA, tal y como se muestra en el tutorial «Si utiliza una capa de compatibilidad con CUDA» aquí: #collapsible -cuda-compat https://docs.aws.amazon.com/sagemaker/latest/dg/inference-gpu-drivers.html

Futuras actualizaciones del sistema operativo TensorFlow

TensorFlow La 2.16 será el último DLAMI que utilice el sistema operativo Ubuntu 20.04. A partir de la TensorFlow versión 2.17 y versiones posteriores, DLamis comenzará a utilizar Ubuntu 22.04 como sistema operativo base. Los clientes que deseen actualizar a estas nuevas versiones deben asegurarse de que sus flujos de trabajo estén preparados para esta actualización.

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

Con la última TF2.16 versión, 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 el uso de Keras 3.0, elimine esta variable de entorno de su entorno TensorFlow virtual/opt/tensorflow mediante el siguiente script:

source /opt/tensorflow/bin/activate unset TF_USE_LEGACY_KERAS

Fecha de lanzamiento: 17-02-2020

Nombres de AMI:

  • Aprendizaje profundo OSS Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20250215

  • Controlador Nvidia AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20250215 (Amazon Linux 2) patentado por Nvidia

Actualizado
Eliminaciones

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

Nombres de AMI:

  • OSS de aprendizaje profundo Nvidia Driver AMI GPU 2.16 (Amazon Linux 2) 20250120 TensorFlow

  • Controlador Nvidia AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20250118 propietario de Deep Learning

Actualizado

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

Nombres de AMI:

  • Controlador Nvidia AMI GPU TensorFlow 2.16 para OSS de aprendizaje profundo (Amazon Linux 2) 20241022

  • Controlador Nvidia propietario de aprendizaje profundo AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20241023

Actualizado

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

Nombres de AMI:

  • Controlador de aprendizaje profundo OSS Nvidia AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240928

  • Controlador Nvidia propietario de aprendizaje profundo AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240928

Actualizado
  • Se actualizó el Nvidia Container Toolkit de la versión 1.16.1 a la 1.16.2 para solucionar la vulnerabilidad de seguridad. CVE-2024-0133

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

Nombres de AMI:

  • Deep Learning OSS Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240921

  • Controlador Nvidia propietario de aprendizaje profundo AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240921

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.

  • PyTorch Versión actualizada de la versión 2.3.0 a la 2.3.1

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

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

Nombres de AMI:

  • Deep Learning OSS Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240817

Added

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

Nombres de AMI:

  • Controlador Nvidia de aprendizaje profundo OSS AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240725

Actualizado
  • Se actualizó la versión TensorFlow del parche de la versión 2.16.1 a la 2.16.2

  • Se ha corregido la versión TensorFlow secundaria incorrecta de DLAMI publicada el 17 de julio de 2022

    • La versión 20240717 de Deep Learning OSS Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) contenía inadvertidamente la versión secundaria 2.17 en lugar de la 2.16. TensorFlow Asegúrese de que los flujos de trabajo basados en la versión 2.16 se actualicen a la versión más reciente de DLAMI. TensorFlow

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

Nombres de AMI:

  • Deep Learning OSS Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240607

  • Controlador Nvidia propietario de Deep Learning AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240610

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

Fecha de lanzamiento: 5-10 de octubre de 2024

Consulte Cambios importantes en las DLAMI.

Nombres de AMI:

  • Controlador Nvidia propietario de Deep Learning AMI GPU TensorFlow 2.16 (Amazon Linux 2) 20240510

  • Controlador Nvidia AMI GPU TensorFlow 2.16 para OSS de aprendizaje profundo (Amazon Linux 2) 20240510

Added
  • Publicación inicial de:

    • Serie Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) (Amazon Linux 2), propietario de Deep Learning.

    • Serie Nvidia Driver AMI GPU TensorFlow 2.16 (Amazon Linux 2) para OSS de aprendizaje profundo.

    • 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 agregó el entorno virtual Tensorflow (fuente del comando de activación//opt/tensorflow). bin/activate Este entorno incluye lo siguiente:

    • “tensorflow=2.16.1”

    • NOTA

      • Para empezar 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 el uso de Keras 3.0, elimine esta variable de entorno de su entorno TensorFlow virtual/opt/tensorflow mediante el siguiente script:

source /opt/tensorflow/bin/activate unset TF_USE_LEGACY_KERAS