Update linux_cuda_wheel.yaml - use manylinux2_28 for testing (#414)

Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
pytorch · Nov 29, 2024 · 7c757be · 7c757be
1 parent f9dd9b5
commit 7c757be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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: