Starten Sie eine GPU-Container-Instance für Amazon ECS

Um eine GPU-Instance auf Amazon ECS zu verwenden, müssen Sie eine Startvorlage und eine Benutzerdatendatei erstellen und die Instance starten.

Anschließend können Sie eine Aufgabe ausführen, die eine für GPU konfigurierte Aufgabendefinition verwendet.

Verwenden einer Startvorlage

Sie können eine Startvorlage erstellen.

• Erstellen Sie eine Startvorlage, die die Amazon ECS-optimierte GPU-AMI-ID für das AMI verwendet. Informationen zum Erstellen einer Startvorlage finden Sie unter Erstellen einer neuen Startvorlage mithilfe von Parametern, die Sie im EC2 Amazon-Benutzerhandbuch definieren.

  Verwenden Sie die AMI-ID aus dem vorherigen Schritt für das Amazon Machine-Image. Informationen zur Angabe der AMI-ID mit dem Systems Manager Manager-Parameter finden Sie unter Angeben eines Systems Manager Manager-Parameters in einer Startvorlage im EC2 Amazon-Benutzerhandbuch.

  Fügen Sie den Benutzerdaten in der Startvorlage Folgendes hinzu. Ersetzen Sie cluster-name mit dem Namen Ihres Clusters.

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

Verwenden Sie die AWS CLI

Sie können den verwenden AWS CLI , um die Container-Instance zu starten.

1. Erstellen Sie eine Datei mit dem Namen userdata.toml. Diese Datei wird für Instance-Benutzerdaten verwendet. Ersetzen Sie cluster-name mit dem Namen Ihres Clusters.

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

2. Führen Sie den folgenden Befehl aus, um die GPU-AMI-ID abzurufen. Sie verwenden dies im folgenden Schritt.

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

3. Führen Sie den folgenden Befehl aus, um die GPU-Instanz zu starten. Denken Sie daran, die folgenden Parameter zu ersetzen:

   • subnetErsetzen Sie es durch die ID des privaten oder öffentlichen Subnetzes, in dem Ihre Instance gestartet wird.

   • gpu_amiErsetzen Sie durch die AMI-ID aus dem vorherigen Schritt.

   • t3.largeErsetzen Sie durch den Instance-Typ, den Sie verwenden möchten.

   • regionErsetzen Sie es durch den Regionalcode.

   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. Führen Sie den folgenden Befehl aus, um zu überprüfen, ob die Container-Instance im Cluster registriert ist. Denken Sie beim Ausführen dieses Befehls daran, die folgenden Parameter zu ersetzen:

   • Ersetzen Sie cluster mit Ihrem Clusternamen.

   • regionErsetzen Sie es durch Ihren Regionalcode.

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