AWS GPU ARM64 AMI de aprendizaje profundo PyTorch 2.6 (Ubuntu 22.04) - AWS Deep Learning AMIs
Formato de nombre AMI EC2 Instancias compatiblesContenido de AMI​AvisosFecha de lanzamiento: 21 de febrero de 2020

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 ARM64 AMI de aprendizaje profundo PyTorch 2.6 (Ubuntu 22.04)

Si necesita ayuda para empezar, consulteIntroducción a las DLAMI.

Formato de nombre AMI

  • Controlador de aprendizaje profundo ARM64 AMI OSS NVIDIA GPU PyTorch 2.6. $ {PATCH-VERSION} (Ubuntu 22.04) $ {AAAA-MM-DD}

EC2 Instancias compatibles

  • G5g

La AMI incluye lo siguiente:

  • AWS Servicio compatible: Amazon EC2

  • Sistema operativo: Ubuntu 22.04

  • Arquitectura de cómputo: ARM64

  • Python:/opt/pytorch/bin/python

  • Versión de Python: 3.12

  • Controlador NVIDIA:

    • Controlador OSS Nvidia: 570.86.15

  • Pila NVIDIA 1.6 CUDA12:

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

    • CUDA predeterminado: 12.6

      • RUTA/-12.6/ usr/local/cuda points to /usr/local/cuda

      • Actualizado a continuación de las variables de entorno:

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

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

    • Sistema compilado (versión NCCL) presente enusr/local/cuda/: 2.24.3

    • PyTorch Versión NCCL compilada del entorno venv: 2.21.5 PyTorch

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

  • Tipo de volumen de EBS: gp3

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

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

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

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

​Avisos

PyTorch Depreciación del canal Anaconda

A partir de la PyTorch versión 2.6, PyTorch ha dejado de ser compatible con Conda (consulte el anuncio oficial). Como resultado, la versión PyTorch 2.6 y las versiones posteriores pasarán a utilizar entornos virtuales de Python. Para activar el PyTorch venv, utilice la fuente/opt/pytorch/bin/activate  

Fecha de lanzamiento: 21 de febrero de 2020

Nombre de AMI: Deep Learning ARM64 AMI OSS NVIDIA Driver GPU PyTorch 2.6.0 (Ubuntu 22.04) 20250221

Added

  • Versión inicial de la serie Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.6.0 (Ubuntu 22.04). Incluye un entorno virtual Python pytorch (source/opt/pytorch/bin/activate), complementado con el controlador NVIDIA R570, CUDA=12.6, CUDNN=9.7, NCCL=2.21.5. PyTorch