AWS GPU ARM64 AMI d'apprentissage profond PyTorch 2.6 (Ubuntu 22.04) - AWS Apprentissage profond (deep learning) AMIs
Format du nom de l'AMI EC2 Instances prises en chargeContenu AMIAvisDate de sortie : 2025-02-21

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

AWS GPU ARM64 AMI d'apprentissage profond PyTorch 2.6 (Ubuntu 22.04)

Pour obtenir de l'aide pour démarrer, consultezCommencer à utiliser le DLAMI.

Format du nom de l'AMI

  • Deep Learning ARM64 AMI OSS NVIDIA Driver GPU PyTorch 2.6. $ {PATCH-VERSION} (Ubuntu 22.04) $ {YYYY-MM-DD}

EC2 Instances prises en charge

  • G5g

L'AMI inclut les éléments suivants :

  • AWS Service pris en charge : Amazon EC2

  • Système d'exploitation : Ubuntu 22.04

  • Architecture informatique : ARM64

  • Python :/opt/pytorch/bin/python

  • Version de Python : 3.12

  • Pilote NVIDIA :

    • Pilote OSS Nvidia : 570.86.15

  • Stack NVIDIA CUDA12 1.6 :

    • Chemin d'installation de CUDA, NCCL et CudDN :/-12.6/ usr/local/cuda

    • CUDA par défaut : 12,6

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

      • Mise à jour des variables d'environnement ci-dessous :

        • LD_LIBRARY_PATH doit avoir/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

        • CHEMIN à avoir/usr/local/cuda/bin/:/usr/local/cuda/include/

    • Version NCCL du système compilé présente à l'adresse/usr/local/cuda/: 2.24.3

    • PyTorch Version NCCL compilée à partir de l'environnement PyTorch venv : 2.21.5

  • AWS CLI v2 en tant qu'aws2 et AWS CLI v1 en tant qu'aws

  • Type de volume EBS : GP3

  • Requête AMI-ID avec le paramètre SSM (exemple de région : 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

  • Requête AMI-ID avec AWSCLI (exemple de région : 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

Avis

PyTorch Obsolète d'Anaconda Channel

À partir de la PyTorch version 2.6, le support de Conda est PyTorch devenu obsolète (voir l'annonce officielle). Par conséquent, les PyTorch versions 2.6 et supérieures utiliseront les environnements virtuels Python. Pour activer le PyTorch venv, veuillez utiliser la source/opt/pytorch/bin/activate  

Date de sortie : 2025-02-21

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

Ajouté

  • Version initiale de la série Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.6.0 (Ubuntu 22.04). Incluant un environnement virtuel Python pytorch (source/opt/pytorch/bin/activate), complété par le pilote NVIDIA R570, CUDA = 12,6, cuDNN = 9,7, NCCL = 2,21,5. PyTorch