diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml index abaa2c1e5..44187151a 100644 --- a/.github/workflows/docs.yaml +++ b/.github/workflows/docs.yaml @@ -57,7 +57,7 @@ jobs: cuda-version: ['12.4'] 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 9fd03457b..36145fdbf 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -69,7 +69,7 @@ jobs: cuda-version: ['11.8', '12.1', '12.4'] 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: