From 89c143d33b9e02faf8d6818a9c47b0d5da68a610 Mon Sep 17 00:00:00 2001
From: Colin Unger <lockshaw@lockshaw.net>
Date: Wed, 18 Dec 2024 15:11:06 -0800
Subject: [PATCH] Add runs-on repo config (#1559)

---
 .github/runs-on.yml | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 .github/runs-on.yml

diff --git a/.github/runs-on.yml b/.github/runs-on.yml
new file mode 100644
index 0000000000..14f75549dd
--- /dev/null
+++ 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