Lancement d’une instance de conteneur GPU pour Amazon ECS

Pour utiliser une instance GPU sur Amazon ECS sur Amazon EC2, vous devez créer un modèle de lancement, un fichier de données utilisateur, puis lancer l'instance.

Vous pouvez ensuite exécuter une tâche qui utilise une définition de tâche configurée pour le GPU.

Utiliser un modèle de lancement

Vous pouvez créer un modèle de lancement.

• Créez un modèle de lancement qui utilise l’ID d’AMI GPU optimisé pour Amazon ECS pour l’AMI. Pour plus d'informations sur la création d'un modèle de lancement, consultez Créer un nouveau modèle de lancement à l'aide des paramètres que vous définissez dans le guide de EC2 l'utilisateur Amazon.

  Utilisez l’ID d’AMI de l’étape précédente pour l’Amazon Machine image. Pour plus d'informations sur la façon de spécifier l'ID AMI avec le paramètre Systems Manager, consultez Spécifier un paramètre Systems Manager dans un modèle de lancement du Amazon EC2 User Guide.

  Ajoutez ce qui suit à la section Données utilisateur du modèle de lancement. Remplacez cluster-name par le nom de votre cluster.

  #!/bin/bash
  echo ECS_CLUSTER=cluster-name >> /etc/ecs/ecs.config;
  echo ECS_ENABLE_GPU_SUPPORT=true >> /etc/ecs/ecs.config

Utilisez le AWS CLI

Vous pouvez utiliser le AWS CLI pour lancer l'instance de conteneur.

1. Créez un fichier appelé userdata.toml. Ce fichier est utilisé pour les données utilisateur de l'instance. Remplacez cluster-name par le nom de votre cluster.

   #!/bin/bash
   echo ECS_CLUSTER=cluster-name >> /etc/ecs/ecs.config;
   echo ECS_ENABLE_GPU_SUPPORT=true >> /etc/ecs/ecs.config

2. Exécutez la commande suivante pour obtenir l’ID d’AMI GPU. Utilisez ceci lors de l'étape suivante.

   aws ssm get-parameters --names /aws/service/ecs/optimized-ami/amazon-linux-2/gpu/recommended --region us-east-1

3. Exécutez la commande suivante pour lancer l’instance GPU. N'oubliez pas de remplacer les paramètres suivants :

   • subnetRemplacez-le par l'ID du sous-réseau privé ou public dans lequel votre instance sera lancée.

   • gpu_amiRemplacez-le par l'ID AMI de l'étape précédente.

   • t3.largeRemplacez-le par le type d'instance que vous souhaitez utiliser.

   • Remplacez region par le code de région.

   aws ec2 run-instances --key-name ecs-gpu-example \
      --subnet-id subnet \
      --image-id gpu_ami \
      --instance-type t3.large \
      --region region \
      --tag-specifications 'ResourceType=instance,Tags=[{Key=GPU,Value=example}]' \
      --user-data file://userdata.toml \
      --iam-instance-profile Name=ecsInstanceRole          

4. Exécutez la commande suivante pour vérifier que l'instance de conteneur est enregistrée auprès du cluster. Lorsque vous exécutez cette commande, n'oubliez pas de remplacer les paramètres suivants :

   • Remplacez cluster par le nom de votre cluster.

   • regionRemplacez-le par votre code de région.

   aws ecs list-container-instances --cluster cluster-name --region region