From 71f28fef571f08ee97a0735077db7e2430861fe8 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 29 Nov 2024 10:45:18 +0000 Subject: [PATCH] Similar changes to doc build job --- .github/workflows/docs.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 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: