From aaa2dcc3b456666ef4facc6970160abbd1224652 Mon Sep 17 00:00:00 2001 From: Won-Kyu Park Date: Thu, 15 Feb 2024 21:18:04 +0900 Subject: [PATCH] use ubuntu20.04 images * fix cmake version: default cmake version of ubuntu20.04 is too low --- .github/workflows/python-package.yml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index 97c1dccda..ebe0c93d6 100644 --- a/.github/workflows/python-package.yml +++ b/.github/workflows/python-package.yml @@ -120,11 +120,12 @@ jobs: [[ "${{ matrix.os }}" = windows-* ]] && python3 -m pip install ninja for NO_CUBLASLT in ON OFF; do if [ ${build_os:0:6} == ubuntu ]; then - image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu22.04 + image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu20.04 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 \ + && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3 python3-pip \ + && pip install cmake==3.27.9 \ && cmake -DCOMPUTE_BACKEND=cuda -DUSE_CUDA_WRAPPER=ON -DNO_CUBLASLT=${NO_CUBLASLT} . \ && cmake --build ." else