cluster-toolkit/pkg/config/accelerators.json at 3a63504fc32ef526b78fded536005797fcfc6572 · GoogleCloudPlatform/cluster-toolkit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
{
  "gpus": {
    "a2-highgpu-1g":              { "count": 1, "type": "nvidia-tesla-a100" },
    "a2-highgpu-2g":              { "count": 2, "type": "nvidia-tesla-a100" },
    "a2-highgpu-4g":              { "count": 4, "type": "nvidia-tesla-a100" },
    "a2-highgpu-8g":              { "count": 8, "type": "nvidia-tesla-a100" },
    "a2-megagpu-16g":             { "count": 16, "type": "nvidia-tesla-a100" },
    "a2-ultragpu-1g":             { "count": 1, "type": "nvidia-a100-80gb" },
    "a2-ultragpu-2g":             { "count": 2, "type": "nvidia-a100-80gb" },
    "a2-ultragpu-4g":             { "count": 4, "type": "nvidia-a100-80gb" },
    "a2-ultragpu-8g":             { "count": 8, "type": "nvidia-a100-80gb" },
    "a3-highgpu-1g":              { "count": 1, "type": "nvidia-h100-80gb" },
    "a3-highgpu-2g":              { "count": 2, "type": "nvidia-h100-80gb" },
    "a3-highgpu-4g":              { "count": 4, "type": "nvidia-h100-80gb" },
    "a3-highgpu-8g":              { "count": 8, "type": "nvidia-h100-80gb" },
    "a3-megagpu-8g":              { "count": 8, "type": "nvidia-h100-mega-80gb" },
    "a3-ultragpu-8g":             { "count": 8, "type": "nvidia-h200-141gb" },
    "a4-highgpu-8g-lowmem":       { "count": 8, "type": "nvidia-b200" },
    "a4-highgpu-8g":              { "count": 8, "type": "nvidia-b200" },
    "a4x-highgpu-4g":             { "count": 4, "type": "nvidia-gb200" },
    "a4x-highgpu-4g-nolssd":      { "count": 4, "type": "nvidia-gb200" },
    "a4x-maxgpu-4g-metal":        { "count": 4, "type": "nvidia-gb300" },
    "a4x-maxgpu-4g-metal-nolssd": { "count": 4, "type": "nvidia-gb300" },
    "g2-standard-4":              { "count": 1, "type": "nvidia-l4" },
    "g2-standard-8":              { "count": 1, "type": "nvidia-l4" },
    "g2-standard-12":             { "count": 1, "type": "nvidia-l4" },
    "g2-standard-16":             { "count": 1, "type": "nvidia-l4" },
    "g2-standard-24":             { "count": 2, "type": "nvidia-l4" },
    "g2-standard-32":             { "count": 1, "type": "nvidia-l4" },
    "g2-standard-48":             { "count": 4, "type": "nvidia-l4" },
    "g2-standard-96":             { "count": 8, "type": "nvidia-l4" },
    "g4-standard-48":             { "count": 1, "type": "nvidia-rtx-pro-6000" },
    "g4-standard-96":             { "count": 2, "type": "nvidia-rtx-pro-6000" },
    "g4-standard-192":            { "count": 4, "type": "nvidia-rtx-pro-6000" },
    "g4-standard-384":            { "count": 8, "type": "nvidia-rtx-pro-6000" }
  },
  "tpus": {
    "ct4p-hightpu-4t":   { "count": 4, "type": "tpu-v4-podslice" },
    "ct5lp-hightpu-1t":  { "count": 1, "type": "tpu-v5-lite-podslice" },
    "ct5lp-hightpu-4t":  { "count": 4, "type": "tpu-v5-lite-podslice" },
    "ct5lp-hightpu-8t":  { "count": 8, "type": "tpu-v5-lite-podslice" },
    "ct5p-hightpu-1t":   { "count": 1, "type": "tpu-v5p-slice" },
    "ct5p-hightpu-2t":   { "count": 2, "type": "tpu-v5p-slice" },
    "ct5p-hightpu-4t":   { "count": 4, "type": "tpu-v5p-slice" },
    "ct6e-standard-1t":  { "count": 1, "type": "tpu-v6e-slice" },
    "ct6e-standard-4t":  { "count": 4, "type": "tpu-v6e-slice" },
    "ct6e-standard-8t":  { "count": 8, "type": "tpu-v6e-slice" },
    "tpu7x-standard-4t": { "count": 4, "type": "tpu7x" }
  }
}