Skip to content

GitLab

Unverified Commit d870f9c5 authored by Matthew Douglas's avatar Matthew Douglas Committed by GitHub
Browse files

Switch CUDA builds to use Rocky Linux 8 container (#1638)

parent 4fb52dc6
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 6 deletions
.github/scripts/build-cuda.sh
View file @ d870f9c5
...@@ -24,13 +24,16 @@ fi ...@@ -24,13 +24,16 @@ fi
[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja [[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
if [ "${build_os:0:6}" == ubuntu ]; then 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" echo "Using image $image"
docker run --platform "linux/$build_arch" -i -w /src -v "$PWD:/src" "$image" sh -c \
"apt-get update \ docker run -i -w /src -v "$PWD:/src" "$image" bash -c \
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \ "dnf update -y \
&& cmake -DPTXAS_VERBOSE=1 -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \ && dnf install cmake gcc-toolset-11 -y \
&& cmake --build ." && source scl_source enable gcc-toolset-11 \
&& cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \
&& cmake --build . --config Release"
else else
pip install cmake==3.28.3 pip install cmake==3.28.3
cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY="${build_capability}" -DCMAKE_BUILD_TYPE=Release -S . cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY="${build_capability}" -DCMAKE_BUILD_TYPE=Release -S .
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment