Add a runner with 8 GPUs for Distributed team (#4931)

Distributed team need a runner with 8 GPUs o test different types of parallelism.
pytorch · Feb 7, 2024 · 598c7b9 · 598c7b9
1 parent dd40947
commit 598c7b9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/scale-config.yml b/.github/scale-config.yml
@@ -75,6 +75,12 @@ runner_types:
     is_ephemeral: false
     max_available: 50
     os: linux
+  linux.g4dn.metal.nvidia.gpu:
+    disk_size: 150
+    instance_type: g4dn.metal
+    is_ephemeral: false
+    max_available: 30
+    os: linux
   linux.g5.12xlarge.nvidia.gpu:
     disk_size: 150
     instance_type: g5.12xlarge