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

Co-authored-by: Nicolas Hug <[email protected]>

Author: atalman

.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:

.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: