Avvio di un'istanza di container GPU per Amazon ECS

Per utilizzare un'istanza GPU su Amazon ECS su Amazon EC2, è necessario creare un modello di avvio, un file di dati utente e avviare l'istanza.

È quindi possibile eseguire un'attività che utilizza una definizione di attività configurata per la GPU.

Utilizzo di un modello di avvio

È possibile creare un modello di avvio.

• Crea un modello di avvio che utilizzi l'ID AMI Amazon ECS-optimized GPU per l'AMI. Per informazioni su come creare un modello di avvio, consultare Create a new launch template using parameters you define nella Guida dell'utente di Amazon EC2.

  Utilizzare l'ID AMI ottenuto nel passaggio precedente per l'immagine Amazon Machine. Per informazioni su come specificare l'ID AMI con il parametro Systems Manager, consultare Specify a Systems Manager parameter in a launch template nella Guida dell'utente di Amazon EC2.

  Aggiungere quanto segue ai dati utente nel modello di avvio. Sostituisci cluster-name con il nome del cluster.

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

Usa il AWS CLI

È possibile utilizzare il AWS CLI per avviare l'istanza del contenitore.

1. Crea un file denominato userdata.toml. Questo file viene utilizzato per i dati utente dell'istanza. Sostituisci cluster-name con il nome del cluster.

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

2. Eseguire questo comando per ottenere l'ID AMI GPU. Ti servirà per la fase successiva.

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

3. Eseguire il comando seguente per avviare l'istanza GPU. Ricordati di sostituire i seguenti parametri:

   • Sostituiscilo subnet con l'ID della sottorete privata o pubblica in cui verrà avviata l'istanza.

   • Sostituisci gpu_ami con l'ID AMI del passaggio precedente.

   • Sostituisci t3.large con il tipo di istanza che desideri utilizzare.

   • Sostituisci region con il codice regionale.

   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. Esegui il comando seguente per verificare che l'istanza di container sia registrata nel cluster. Quando esegui questo comando, ricordati di sostituire i parametri seguenti:

   • Sostituisci cluster con il nome del cluster.

   • Sostituisci region con il tuo codice regionale.

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