From 30966e3bc9fdc276a121e2b300a8d80e1ea12c0a Mon Sep 17 00:00:00 2001
From: Kevin Lin <42618777+kevin314@users.noreply.github.com>
Date: Wed, 9 Apr 2025 12:16:05 -0500
Subject: [PATCH] [CI] Set allowedCudaVersions (#329)

---
 .github/scripts/runpod_api.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/scripts/runpod_api.py b/.github/scripts/runpod_api.py
index 9836caaba..9ebdd0b04 100644
--- a/.github/scripts/runpod_api.py
+++ b/.github/scripts/runpod_api.py
@@ -53,7 +53,8 @@ def create_pod():
         "volumeInGb": args.volume_size,
         "gpuTypeIds": [args.gpu_type],
         "gpuCount": args.gpu_count,
-        "imageName": args.image
+        "imageName": args.image,
+        "allowedCudaVersions": ["12.4"]
     }
 
     response = requests.post(PODS_API, headers=HEADERS, json=payload)
@@ -170,7 +171,7 @@ def execute_command(pod_id):
                                    stdout=subprocess.PIPE,
                                    stderr=subprocess.STDOUT,
                                    universal_newlines=True,
-                                   bufsize=1)
+                                   bufsize=0)
 
         stdout_lines = []