From 7101f3f074faf238fff92b302059460d66080ebb Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 18 Nov 2024 16:36:13 +0000 Subject: [PATCH] Use manylinux2_28 for testing CUDA wheels --- .github/workflows/docs.yaml | 2 +- .github/workflows/linux_cuda_wheel.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml index abaa2c1e5..44187151a 100644 --- a/.github/workflows/docs.yaml +++ b/.github/workflows/docs.yaml @@ -57,7 +57,7 @@ jobs: cuda-version: ['12.4'] ffmpeg-version-for-tests: ['7'] container: - image: "pytorch/manylinux-builder:cuda${{ matrix.cuda-version }}" + image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}" options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility" needs: build steps: diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 9fd03457b..36145fdbf 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -69,7 +69,7 @@ jobs: cuda-version: ['11.8', '12.1', '12.4'] ffmpeg-version-for-tests: ['5', '6', '7'] container: - image: "pytorch/manylinux-builder:cuda${{ matrix.cuda-version }}" + image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}" options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility" needs: build steps: