Créez une file d'attente SageMaker de tâches de formation dans AWS Batch

SageMaker Les files d'attente pour les tâches de formation s'intègrent directement au service d' SageMaker intelligence artificielle pour permettre une planification des tâches sans serveur sans que vous ayez à gérer l'infrastructure informatique sous-jacente.

Prérequis

Avant de créer une file d'attente de tâches de SageMaker formation, assurez-vous d'avoir :

• Environnement de service : environnement de service qui définit les limites de capacité. Pour de plus amples informations, veuillez consulter Créez un environnement de service dans AWS Batch.

• Autorisations IAM : autorisations permettant de créer et de gérer des files AWS Batch d'attente de travail et des environnements de service. Pour de plus amples informations, veuillez consulter AWS Batch Politiques, rôles et autorisations IAM.

Create a SageMaker Training job queue (AWS Batch console)

1. Ouvrez la AWS Batch console à l'adresse https://console.aws.amazon.com/batch/.

2. Dans le volet de navigation, choisissez Job queues puis Create.

3. Pour le type d'orchestration, choisissez SageMaker Entraînement.

4. Pour la configuration de la file d'attente des Job :

   1. Dans Nom, entrez le nom de la file d'attente de Job.

   2. pour Priorité, entrez une valeur comprise entre 0 et 1000. Une file d'attente de Job avec une priorité plus élevée est privilégiée pour les environnements de service.

   3. (Facultatif) Pour la politique de planification Amazon Resource Name (ARN), choisissez une politique de planification existante.

   4. Pour les environnements de service connectés, sélectionnez un environnement de service dans la liste à associer à la file d'attente des tâches.

5. (Facultatif) Pour les limites d'état des Job :

   1. Pour Mauvaise configuration, choisissez SERVICE_ENVIRONMENT_MAX_RESOURCE et entrez la durée maximale d'exécution (secondes).

   2. Pour Capacité, choisissez INSUFFICIENT_INSTANCE_CAPACITY et entrez la durée maximale d'exécution (secondes).

6. Choisissez Créer une file d'attente de tâches

Create a SageMaker Training job queue (AWS CLI)

Utilisez la create-job-queue commande pour créer une file d'attente SageMaker de tâches de formation.

L'exemple suivant crée une file d'attente de tâches de SageMaker formation de base qui utilise un environnement de service :

aws batch create-job-queue \
  --job-queue-name my-sm-training-fifo-jq \
  --job-queue-type SAGEMAKER_TRAINING \
  --priority 1 \
  --service-environment-order order=1,serviceEnvironment=ExampleServiceEnvironment

ExampleServiceEnvironmentRemplacez-le par le nom de votre environnement de service.

La commande renvoie un résultat semblable à ce qui suit :

{
  "jobQueueName": "my-sm-training-fifo-jq",
  "jobQueueArn": "arn:aws:batch:region:account:job-queue/my-sm-training-fifo-jq"
}

Après avoir créé votre file d'attente de tâches, vérifiez qu'elle a été créée avec succès et qu'elle est dans un état valide.

Utilisez la describe-job-queues commande pour afficher les détails de votre file d'attente de tâches :

aws batch describe-job-queues --job-queues my-sm-training-fifo-jq

La commande renvoie un résultat semblable à ce qui suit :

{
  "jobQueues": [
    {
      "jobQueueName": "my-sm-training-fifo-jq",
      "jobQueueArn": "arn:aws:batch:region:account:job-queue/my-sm-training-fifo-jq",
      "state": "ENABLED",
      "status": "VALID",
      "statusReason": "JobQueue Healthy",
      "priority": 1,
      "computeEnvironmentOrder": [],
      "serviceEnvironmentOrder": [
        {
          "order": 1,
          "serviceEnvironment": "arn:aws:batch:region:account:service-environment/ExampleServiceEnvironment"
        }
      ],
      "jobQueueType": "SAGEMAKER_TRAINING",
      "tags": {},
      "jobStateTimeLimitActions": []
    }
  ]
}

Assurez-vous que :

• Le state est ENABLED

• Le status est VALID

• Le statusReason est JobQueue Healthy

• Le jobQueueType est SAGEMAKER_TRAINING

• Les serviceEnvironmentOrder références de votre environnement de service