From a157bb8907378845f661f369b1828493daa4844f Mon Sep 17 00:00:00 2001 From: Fred Heinecke Date: Tue, 21 Jul 2026 17:00:37 -0500 Subject: [PATCH] [CI] Fix old NVCC dependency in GH CI Signed-off-by: Fred Heinecke --- .github/workflows/build.yml | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index fe69a3ace2..a99000c2e0 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -16,13 +16,16 @@ jobs: name: 'Core' runs-on: ubuntu-latest container: - image: nvcr.io/nvidia/cuda:12.1.0-devel-ubuntu22.04 + image: nvcr.io/nvidia/cuda:12.9.2-devel-ubuntu22.04 options: --user root steps: - name: 'Dependencies' run: | apt-get update - apt-get install -y git python3.9 pip cudnn9-cuda-12 + apt-mark unhold libnccl2 libnccl-dev + apt-get install -y git python3 python3-pip cudnn9-cuda-12 \ + libnccl2=2.30.7-1+cuda12.9 \ + libnccl-dev=2.30.7-1+cuda12.9 pip install cmake==3.21.0 pybind11[global] ninja "nvidia-cudnn-frontend>=1.25.0" - name: 'Checkout' uses: actions/checkout@v3 @@ -70,13 +73,16 @@ jobs: - name: Start named container run: | - docker run -v $(pwd):$(pwd) -w $(pwd) --name builder -d nvcr.io/nvidia/cuda:12.8.0-devel-ubuntu22.04 sleep infinity + docker run -v $(pwd):$(pwd) -w $(pwd) --name builder -d nvcr.io/nvidia/cuda:12.9.2-devel-ubuntu22.04 sleep infinity - name: 'Dependencies' run: | docker exec builder bash -c '\ apt-get update && \ - apt-get install -y git python3.9 pip cudnn9-cuda-12 && \ + apt-mark unhold libnccl2 libnccl-dev && \ + apt-get install -y git python3 python3-pip cudnn9-cuda-12 \ + libnccl2=2.30.7-1+cuda12.9 \ + libnccl-dev=2.30.7-1+cuda12.9 && \ pip install cmake torch ninja pydantic importlib-metadata>=1.0 packaging pybind11 numpy einops onnxscript "nvidia-cudnn-frontend>=1.25.0" && \ apt-get clean \ '