AWS GPU ARM64 AMI PyTorch 2.6 de aprendizaje profundo (Amazon Linux 2023) - 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 PyTorch 2.6 de aprendizaje profundo (Amazon Linux 2023)

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} (Amazon Linux 2023) $ {AAAA-MM-DD}

EC2 Instancias compatibles

  • G5g

La AMI incluye lo siguiente:

  • AWS Servicio compatible: EC2

  • Sistema operativo: Amazon Linux 2023

  • Arquitectura informática: ARM64

  • Python:/opt/pytorch/bin/python

  • Versión de Python: 3.12

  • Controlador NVIDIA:

    • Controlador OSS Nvidia: 570.86.15

  • CUDA12Pila NVIDIA 1.6:

    • 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 conda: 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-amazon-linux-2023/latest/ami-id  \
    --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 AMI OSS Nvidia Driver GPU PyTorch 2.6.? (Amazon Linux 2023) ????????' '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 (Amazon Linux 2023) 20250221

Added

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