From 7c757bee3b3f5e034f7f3aaa3eddcb3f52714fce Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Fri, 29 Nov 2024 10:45:32 +0000 Subject: [PATCH] Update linux_cuda_wheel.yaml - use manylinux2_28 for testing (#414) Co-authored-by: Nicolas Hug --- .github/workflows/docs.yaml | 4 ++-- .github/workflows/linux_cuda_wheel.yaml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml index 54019ab5..ac4a8103 100644 --- a/.github/workflows/docs.yaml +++ b/.github/workflows/docs.yaml @@ -54,10 +54,10 @@ jobs: # the wheel unless the label cliflow/binaries/all is present in the # PR. python-version: ['3.9'] - cuda-version: ['12.4'] + cuda-version: ['12.6'] 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 17538d27..a4b001c2 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -66,10 +66,10 @@ jobs: # For the actual release we should add that label and change this to # include more python versions. python-version: ['3.9'] - cuda-version: ['11.8', '12.1', '12.4'] + cuda-version: ['12.6'] 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: