From bc81c4848eebde00275de367aacb7efc8b735e9a Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Thu, 16 Apr 2026 09:40:35 -0400 Subject: [PATCH] DO NOT MERGE: use `citestwheel` image built on `cuda-base` instead of `cuda-devel` --- .github/workflows/wheels-test.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/wheels-test.yaml b/.github/workflows/wheels-test.yaml index d799aef1..a706c025 100644 --- a/.github/workflows/wheels-test.yaml +++ b/.github/workflows/wheels-test.yaml @@ -234,7 +234,7 @@ jobs: runs-on: "linux-${{ matrix.ARCH }}-gpu-${{ matrix.GPU }}-${{ matrix.DRIVER }}-1" continue-on-error: ${{ inputs.continue-on-error }} container: - image: "rapidsai/citestwheel:26.06-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + image: "rapidsai/staging:citestwheel-400-26.06-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" options: ${{ inputs.container-options }} env: NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }} # GPU jobs must set this container env variable