File tree Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Original file line number Diff line number Diff line change 2424[[ " ${build_os} " = windows-* ]] && python3 -m pip install ninja
2525
2626if [ " ${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"
3437else
3538 pip install cmake==3.28.3
3639 cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=" ${build_capability} " -DCMAKE_BUILD_TYPE=Release -S .
You can’t perform that action at this time.
0 commit comments