Add runs-on repo config

flexflow · Dec 18, 2024 · f91904d · f91904d
1 parent 1d5140d
commit f91904d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/.github/runs-on.yml b/.github/runs-on.yml
@@ -0,0 +1,11 @@
+images:
+  dlami-x64:
+    platform: "linux"
+    arch: "x64"
+    owner: "898082745236" # AWS
+    name: "Deep Learning Base OSS Nvidia Driver GPU AMI (Ubuntu 22.04)*"
+
+runners:
+  gpu-nvidia:
+    family: ["g4dn.xlarge"]
+    image: dlami-x64