Avvio di un'istanza di container GPU per Amazon ECS

Per utilizzare un'istanza GPU su Amazon ECS su Amazon EC2, devi 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.

• Creare un modello di avvio che utilizzi l'ID AMI GPU ottimizzato per Amazon ECS per l'AMI. Per informazioni su come creare un modello di lancio, consulta Creare un nuovo modello di lancio utilizzando i parametri definiti nella Amazon EC2 User Guide.

  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, consulta Specificare un parametro Systems Manager in un modello di avvio nella Amazon EC2 User Guide.

  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