AWS AMI de GPU de ARM64 base de aprendizaje profundo (Amazon Linux 2)

• ARM64 Base de aprendizaje profundo OSS Controlador Nvidia GPU AMI (Amazon Linux 2) $ {YYY-MM-DD}

• G5g

• AWS Servicio compatible: Amazon EC2

• Sistema operativo: Amazon Linux 2

• Arquitectura de cómputo: ARM64

• Núcleo de Linux: 5.10

• Controlador NVIDIA: 550.144.03

• Pila NVIDIA CUDA 12.1, 12.2, 12.3:

  • Directorios de instalación de CUDA, NCCL y CUDDN:

    • Ejemplo:/usr/local/cuda-12.1/ , /usr/local/cuda-12.1/

  • Versión NCCL compilada:

    • Para el directorio CUDA de 12.3, compiló la versión 2.21.5+ .4 de NCCL. CUDA12

    • Para el directorio CUDA 12.1, 12.2, compiló la versión 1.18.5+ 2 de NCCL CUDA12

  • CUDA predeterminado: 12.1

    • PATH/usr/local/cudaapunta a CUDA 12.1

    • Se actualizaron las siguientes variables de entorno:

      • LD_LIBRARY_PATH tendrá/64 usr/local/cuda-12.1/lib:/usr/local/cuda-12.1/lib64:/usr/local/cuda-12.1:/usr/local/cuda-12.1/targets/sbsa-linux/lib:/usr/local/cuda-12.1/nvvm/lib64:/usr/local/cuda-12.1/extras/CUPTI/lib

      • PATH para tener//usr/local/cuda-12.1/bin/:/usr/local/cuda-12.1/include

      • Para cualquier versión de CUDA diferente, actualice LD_LIBRARY_PATH en consecuencia.

• AWS CLI v2 en/2 y v1 en/usr/local/bin/aws AWS CLI usr/bin/aws

• Tipo de volumen de EBS: gp3

• Kit de herramientas de contenedores Nvidia: 1.16.2

  • Comando de versión: -V nvidia-container-cli

• Docker: 26.1.2

• Python:/usr/bin/python3.10

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

  SSM_PARAMETER=base-oss-nvidia-driver-gpu-amazon-linux-2/latest/ami-id \
      aws ssm get-parameter --region us-east-1 \
      --name /aws/service/deeplearning/ami/arm64/$SSM_PARAMETER  \
      --query "Parameter.Value" \
      --output text

• Consulta el AMI-ID con AWSCLI (por ejemplo, la región es us-east-1):

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

Kernel

• La versión del núcleo se fija mediante el comando:

  sudo yum versionlock kernel*

• Recomendamos a los usuarios que eviten actualizar su versión del núcleo (a menos que sea debido a un parche de seguridad) para garantizar la compatibilidad con los controladores y las versiones de los paquetes instalados. Si los usuarios aún desean actualizar, pueden ejecutar los siguientes comandos para desfijar sus versiones del núcleo:

  sudo yum versionlock delete kernel*
  sudo yum update -y

• Para cada nueva versión de DLAMI, se utiliza el último núcleo compatible disponible.

• Se actualizó el kit de herramientas NVIDIA Container de la versión 1.17.3 a la versión 1.17.4

  • Consulte la página de notas de la versión aquí para obtener más información:/1.17.4 https://github.com/NVIDIA/ nvidia-container-toolkit releases/tag/v

  • En la versión 1.17.4 del Container Toolkit, el montaje de bibliotecas compatibles con CUDA ahora está deshabilitado. 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 las bibliotecas de compatibilidad con CUDA, tal y como se muestra en el tutorial sobre si utiliza una capa de compatibilidad CUDA.

• Se eliminaron las bibliotecas de espacio de usuario cuobj y nvdisasm proporcionadas por el kit de herramientas CUDA de NVIDIA para abordar las presentes en el boletín de seguridad del kit de herramientas NVIDIA CUDA del 18 de febrero de 2025 CVEs

• Se actualizó el controlador Nvidia de la versión 550.127.05 a la 550.144.03 para cumplir con lo indicado en el boletín de seguridad del controlador de pantalla de CVEs GPU de NVIDIA de enero de 2025

• Se actualizó el controlador Nvidia de la versión 550.90.07 a la 550.127.05 para adaptarlo a lo indicado en el boletín de seguridad de las pantallas de GPU de NVIDIA de CVEs octubre de 2024

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

• Se actualizó la versión del controlador Nvidia a 535.183.01 desde 535.161.08

• Versión inicial de la DLAMI OSS DLAMI de Deep Learning ARM64 Base para Amazon Linux 2