Spécification GPUs dans une définition de ECS tâche Amazon - Amazon Elastic Container Service

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Spécification GPUs dans une définition de ECS tâche Amazon

Pour utiliser l'instance GPUs sur un conteneur et l'GPUexécution de Docker, assurez-vous de désigner le nombre de conteneurs nécessaires à GPUs votre conteneur dans la définition de tâche. Une fois les conteneurs GPUs supportant placés, l'agent de ECS conteneur Amazon épinglera le nombre souhaité de conteneurs physiques GPUs au bon conteneur. Le nombre de conteneurs GPUs réservés pour tous les conteneurs dans une tâche ne peut pas dépasser le nombre de conteneurs disponibles GPUs sur l'instance de conteneur sur laquelle la tâche est lancée. Pour de plus amples informations, veuillez consulter Création d'une définition de ECS tâche Amazon à l'aide de la console.

Important

Si vos GPU exigences ne sont pas spécifiées dans la définition de tâche, la tâche utilise l'exécution par défaut du Docker.

L'exemple suivant illustre le JSON format des GPU exigences dans une définition de tâche :

{
  "containerDefinitions": [
     {
        ...
        "resourceRequirements" : [
            {
               "type" : "GPU", 
               "value" : "2"
            }
        ],
     },
...
}

L'exemple suivant illustre la syntaxe d'un conteneur Docker qui spécifie une GPU exigence. Ce conteneur en utilise 2GPUs, exécute l'nvidia-smiutilitaire, puis se ferme.

{
  "containerDefinitions": [
    {
      "memory": 80,
      "essential": true,
      "name": "gpu",
      "image": "nvidia/cuda:11.0.3-base",
      "resourceRequirements": [
         {
           "type":"GPU",
           "value": "2"
         }
      ],
      "command": [
        "sh",
        "-c",
        "nvidia-smi"
      ],
      "cpu": 100
    }
  ],
  "family": "example-ecs-gpu"
}