AWS GPU ARM64 AMI PyTorch 2.6 pour apprentissage profond (Amazon Linux 2023) - 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 PyTorch 2.6 pour apprentissage profond (Amazon Linux 2023)

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

EC2 Instances prises en charge

  • G5g

L'AMI inclut les éléments suivants :

  • AWS Service pris en charge : EC2

  • Système d'exploitation : Amazon Linux 2023

  • 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 conda : 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 : la région est 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

  • Interrogez l'AMI-ID avec AWSCLI (par exemple, la région est 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

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 (Amazon Linux 2023) 20250221

Ajouté

  • Version initiale de la série Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.6.0 (Amazon Linux 2023). 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