Skip to content

Commit d870f9c

Browse files
Switch CUDA builds to use Rocky Linux 8 container (#1638)
1 parent 4fb52dc commit d870f9c

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

.github/scripts/build-cuda.sh

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,16 @@ fi
2424
[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
2525

2626
if [ "${build_os:0:6}" == ubuntu ]; then
27-
image=nvidia/cuda:${cuda_version}-devel-ubuntu22.04
27+
# We'll use Rocky Linux 8 in order to maintain manylinux 2.24 compatibility.
28+
image="nvidia/cuda:${cuda_version}-devel-rockylinux8"
2829
echo "Using image $image"
29-
docker run --platform "linux/$build_arch" -i -w /src -v "$PWD:/src" "$image" sh -c \
30-
"apt-get update \
31-
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
32-
&& cmake -DPTXAS_VERBOSE=1 -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \
33-
&& cmake --build ."
30+
31+
docker run -i -w /src -v "$PWD:/src" "$image" bash -c \
32+
"dnf update -y \
33+
&& dnf install cmake gcc-toolset-11 -y \
34+
&& source scl_source enable gcc-toolset-11 \
35+
&& cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \
36+
&& cmake --build . --config Release"
3437
else
3538
pip install cmake==3.28.3
3639
cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY="${build_capability}" -DCMAKE_BUILD_TYPE=Release -S .

0 commit comments

Comments
 (0)