From cc913108139c108e35099386639588aebf880681 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Tue, 13 May 2025 15:45:59 -0400
Subject: [PATCH] Switch CUDA builds to use Rocky Linux 8 container

---
 .github/scripts/build-cuda.sh | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/.github/scripts/build-cuda.sh b/.github/scripts/build-cuda.sh
index d8c4ebbe1..8985327f2 100644
--- a/.github/scripts/build-cuda.sh
+++ b/.github/scripts/build-cuda.sh
@@ -24,13 +24,16 @@ fi
 [[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
 
 if [ "${build_os:0:6}" == ubuntu ]; then
-    image=nvidia/cuda:${cuda_version}-devel-ubuntu22.04
+    # We'll use Rocky Linux 8 in order to maintain manylinux 2.24 compatibility.
+    image="nvidia/cuda:${cuda_version}-devel-rockylinux8"
     echo "Using image $image"
-    docker run --platform "linux/$build_arch" -i -w /src -v "$PWD:/src" "$image" sh -c \
-        "apt-get update \
-    && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
-    && cmake -DPTXAS_VERBOSE=1 -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \
-    && cmake --build ."
+
+    docker run -i -w /src -v "$PWD:/src" "$image" bash -c \
+        "dnf update -y \
+        && dnf install cmake gcc-toolset-11 -y \
+        && source scl_source enable gcc-toolset-11 \
+        && cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \
+        && cmake --build . --config Release"
 else
     pip install cmake==3.28.3
     cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY="${build_capability}" -DCMAKE_BUILD_TYPE=Release -S .