Add AWS linux AMI to runs-on for testing (#1589)

flexflow · Jan 30, 2025 · c116872 · c116872
1 parent de7fa32
commit c116872
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/.github/runs-on.yml b/.github/runs-on.yml
@@ -1,4 +1,10 @@
 images:
+  amazon-linux-gpu-x64:
+    platform: "linux"
+    arch: "x64"
+    owner: "898082745236" # AWS
+    name: "Amazon Linux 2 AMI with NVIDIA TESLA GPU Driver*"
+
   dlami-x64:
     platform: "linux"
     arch: "x64"
@@ -8,4 +14,4 @@ images:
 runners:
   gpu-nvidia:
     family: ["g4dn.xlarge"]
-    image: dlami-x64
+    image: amazon-linux-gpu-x64