Similar changes to doc build job

pytorch · Nov 29, 2024 · 71f28fe · 71f28fe
1 parent d1f2169
commit 71f28fe
Showing 1 changed file with 2 additions and 2 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: