Menentukan GPUs dalam definisi ECS tugas Amazon - Amazon Elastic Container Service

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Menentukan GPUs dalam definisi ECS tugas Amazon

Untuk menggunakan instans kontainer GPUs pada kontainer dan GPU waktu aktif Docker, pastikan Anda menetapkan jumlah kontainer yang diperlukan kontainer dalam ketentuan tugas. GPUs Sebagai kontainer yang mendukung GPUs ditempatkan, agen ECS kontainer Amazon pin jumlah fisik yang diinginkan GPUs ke kontainer yang sesuai. Jumlah penampung yang GPUs disimpan untuk semua kontainer dalam tugas tidak dapat melebihi jumlah yang tersedia GPUs pada instans kontainer tempat tugas diluncurkan. Untuk informasi selengkapnya, lihat Membuat definisi ECS tugas Amazon menggunakan konsol.

penting

Jika GPU persyaratan Anda tidak ditentukan dalam ketentuan tugas, tugas akan menggunakan waktu aktif Docker default.

Berikut ini menunjukkan JSON format GPU persyaratan dalam ketentuan tugas:

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

Contoh berikut menunjukkan sintaksis untuk kontainer Docker yang menentukan persyaratan. GPU Kontainer ini menggunakan duaGPUs, menjalankan nvidia-smi utilitas, dan kemudian keluar.

{
  "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"
}