View a markdown version of this page

Utilisation de PCS-ready DLAMI avec AWS PIÈCES - AWS PIÈCES

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.

Utilisation de PCS-ready DLAMI avec AWS PIÈCES

L'AMI GPU de base AWS PCS-ready DLAMI (Ubuntu 24.04) est une image machine AWS-maintained Amazon destinée à l'exécution AI/ML et aux charges de travail HPC sur les PC. AWS Il fournit une base prête pour la production qui vous permet de déployer des clusters en quelques minutes au lieu de créer et de valider des AMI personnalisées.

Ce qui est inclus

PCS-ready DLAMI est basé sur l'AMI GPU Deep Learning Base (Ubuntu 24.04) et ajoute les composants PCS suivants : AWS

  • PCS Agent — L'agent de gestion du cluster AWS PCS

  • Slurm pour AWS PCS — Plusieurs versions de Slurm prises en charge sont préinstallées. La version correcte est activée automatiquement lors du lancement de l'instance en fonction de la configuration de votre cluster.

  • Utilitaires EFS — Pour monter des systèmes de fichiers Amazon EFS

Le DLAMI source fournit le système d'exploitation (Ubuntu 24.04), les pilotes GPU NVIDIA, la boîte à outils CUDA, les pilotes EFA, le client Lustre et d'autres infrastructures de base. Pour plus de détails sur ces composants, consultez les notes de mise à jour de Deep Learning AMI.

PCS-ready Le DLAMI est disponible pour les architectures x86_64 et arm64.

Note

PCS-ready Le DLAMI n'inclut pas les logiciels d'application tels que AI/ML les frameworks (PyTorch, TensorFlow, JAX), les compilateurs ou les bibliothèques mathématiques. Vous pouvez ajouter votre couche d'application sur des systèmes de fichiers partagés ou en créant une AMI personnalisée au-dessus de PCS-ready DLAMI.

Le champ Description de chaque AMI résume son contenu, y compris la source DLAMI sur laquelle elle est basée, la version de l'agent PCS, les versions de Slurm prises en charge et la version des utilitaires EFS. Vous pouvez consulter ce champ dans la console Amazon EC2 ou à l'aide de l'describe-imagesAPI. Voici un exemple de valeur de champ Description :

PCS-Ready DLAMI based on Deep Learning Base OSS Nvidia Driver GPU AMI (Ubuntu 24.04) 20260522. PCS Agent: 1.4.0-1. Slurm: 24.11.7-1, 25.05.7-1, 25.11.2-1. EFS Utils: 2.4.2

Trouvez le PCS-ready DLAMI actuel

Console de gestion AWS
Pour rechercher PCS-ready DLAMI dans la console
  1. Ouvrez la console AWS PCS et naviguez pour créer ou modifier un groupe de nœuds de calcul.

  2. Dans la section de sélection des AMI, sélectionnez PCS-ready AMI.

  3. Une liste déroulante affiche les PCS-ready DLAMI disponibles filtrés selon l'architecture de type d'instance que vous avez sélectionnée.

  4. Choisissez l'AMI de base AWS PCS-ready DLAMI (Ubuntu 24.04). La liste déroulante affiche l'ID AMI et le nom complet de l'AMI ci-dessous à titre de référence.

AWS CLI

Vous pouvez récupérer le dernier ID AMI PCS-ready DLAMI à l'aide du magasin de paramètres Amazon EC2 Systems Manager. Remplacez region-code par votre Région AWS.

  • x86_64

    aws ssm get-parameter --region region-code \ --name /aws/service/pcs/ami/dlami-base-ubuntu2404/x86_64/latest/ami-id \ --query "Parameter.Value" --output text
  • bras 64

    aws ssm get-parameter --region region-code \ --name /aws/service/pcs/ami/dlami-base-ubuntu2404/arm64/latest/ami-id \ --query "Parameter.Value" --output text

Vous pouvez également rechercher PCS-ready DLAMI par modèle de nom :

  • x86_64

    aws ec2 describe-images --region region-code --owners amazon \ --filters 'Name=name,Values=aws-pcs-ready-dlami-base-ubuntu2404-x86_64-*' \ 'Name=state,Values=available' \ --query 'sort_by(Images, &CreationDate)[-1].[Name,ImageId]' --output text
  • bras 64

    aws ec2 describe-images --region region-code --owners amazon \ --filters 'Name=name,Values=aws-pcs-ready-dlami-base-ubuntu2404-arm64-*' \ 'Name=state,Values=available' \ --query 'sort_by(Images, &CreationDate)[-1].[Name,ImageId]' --output text

Utilisez l'ID AMI lorsque vous créez ou mettez à jour un groupe de nœuds de calcul.

Recevez des notifications pour les nouvelles versions

Vous pouvez recevoir des notifications chaque fois qu'un nouveau PCS-ready DLAMI est publié. Les notifications sont publiées sur Amazon SNS à l'aide de la rubrique suivante :

arn:aws:sns:us-west-2:265886551188:pcs-ready-dlami-release-notifications

Les messages sont publiés sur cette rubrique lorsqu'un nouveau PCS-ready DLAMI est publié.

Pour vous abonner aux notifications de PCS-ready publication du DLAMI
  1. Ouvrez la console Amazon SNS.

  2. Dans la barre de navigation, choisissez Région AWS US West (Oregon), si nécessaire. Vous devez sélectionner la région dans laquelle la notification SNS à laquelle vous êtes abonné a été créée.

  3. Dans le volet de navigation, choisissez Abonnements, puis sélectionnez Créer un abonnement.

  4. Dans la boîte de dialogue Créer un abonnement, exécutez l’une des actions suivantes :

    1. Pour Topic ARN, entrez les informations suivantes : arn:aws:sns:us-west-2:265886551188:pcs-ready-dlami-release-notifications

    2. Pour Protocole, choisissez E-mail.

    3. Pour Endpoint, entrez l'adresse e-mail à laquelle vous souhaitez recevoir des notifications.

    4. Choisissez Créer un abonnement.

  5. Vous recevrez un e-mail de confirmation avec pour objet AWS Notification - Confirmation d'abonnement. Ouvrez l’e-mail et choisissez Confirm subscription (Confirmer l’abonnement) pour terminer votre abonnement.

Utilisation avec Infrastructure as Code

Le chemin du paramètre SSM fournit une référence stable qui renvoie toujours au dernier ID AMI. Vous pouvez l'utiliser dans les CloudFormation modèles pour récupérer automatiquement les nouvelles versions lors du redéploiement :

AmiId: '{{resolve:ssm:/aws/service/pcs/ami/dlami-base-ubuntu2404/x86_64/latest/ami-id}}'

Passez à une nouvelle version

AWS publie des versions PCS-ready DLAMI mises à jour lorsque l'AMI GPU Deep Learning Base source est mise à jour ou lorsque des composants PCS (PCS Agent ou Slurm pour PCS) sont mis à jour. Pour mettre à jour votre cluster, récupérez le dernier ID AMI à l'aide du paramètre SSM ou de la recherche de nom décrite ci-dessus, puis mettez à jour chaque groupe de nœuds de calcul pour référencer le nouvel ID AMI.