From 4e44e9cd8e91da65cd97dec33c719f525633c472 Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Thu, 16 Apr 2026 09:40:35 -0400 Subject: [PATCH 1/2] 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 38492090..e5e2907a 100644 --- a/.github/workflows/wheels-test.yaml +++ b/.github/workflows/wheels-test.yaml @@ -165,7 +165,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.08-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 From 46c9aefd791ff3c4ecc4c29c8ce99ccf6b4e7399 Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Tue, 30 Jun 2026 15:57:16 -0400 Subject: [PATCH 2/2] chore: update image for 26.08 --- .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 e5e2907a..4eb6e8af 100644 --- a/.github/workflows/wheels-test.yaml +++ b/.github/workflows/wheels-test.yaml @@ -165,7 +165,7 @@ jobs: runs-on: "linux-${{ matrix.ARCH }}-gpu-${{ matrix.GPU }}-${{ matrix.DRIVER }}-1" continue-on-error: ${{ inputs.continue-on-error }} container: - image: "rapidsai/staging:citestwheel-400-26.06-cuda${{ matrix.CUDA_VER }}-${{ matrix.LINUX_VER }}-py${{ matrix.PY_VER }}" + image: "rapidsai/staging:citestwheel-408-26.08-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