ci: Option to set number of Nvidia GPUs in ARC (#5010)

pytorch · Mar 14, 2024 · 5a9c77e · 5a9c77e
1 parent 15751f1
commit 5a9c77e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/arc-runner-config.yaml b/.github/arc-runner-config.yaml
@@ -20,6 +20,7 @@ runnerConfig:
     memory: 16Gi
     minRunners: 0
     nodeType: compute-gpu
+    nvidiaGpus: 1
     runnerLabel: linux-gpu.large
     containerMode: dind-rootless
     workingDiskSpace: 150Gi