From 04c867066b258190e0d0c61ced26e3ce66857a64 Mon Sep 17 00:00:00 2001 From: Pearu Peterson Date: Mon, 24 Nov 2025 20:33:09 +0200 Subject: [PATCH] Update CUDA version in GH workflows to 2.9. --- .github/scripts/unittest-windows/setup_env.sh | 2 +- .github/workflows/docs.yaml | 6 +++--- .github/workflows/unittest-linux-gpu.yml | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/scripts/unittest-windows/setup_env.sh b/.github/scripts/unittest-windows/setup_env.sh index efea89f548..34dd3d66be 100644 --- a/.github/scripts/unittest-windows/setup_env.sh +++ b/.github/scripts/unittest-windows/setup_env.sh @@ -36,4 +36,4 @@ fi conda activate "${env_dir}" # 3. Install minimal build tools -conda install -y -c conda-forge cmake ninja "ffmpeg<8" +conda install -y -c conda-forge cmake ninja "ffmpeg<9" diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml index 666b6783b6..981ea2ba52 100644 --- a/.github/workflows/docs.yaml +++ b/.github/workflows/docs.yaml @@ -14,15 +14,15 @@ jobs: strategy: matrix: python-version: ["3.11"] - cuda-version: ["12.6"] - ffmpeg-version: ["7"] + cuda-version: ["12.9"] + ffmpeg-version: ["8"] fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main permissions: id-token: write contents: read with: - runner: linux.4xlarge.nvidia.gpu + runner: linux.g5.4xlarge.nvidia.gpu repository: pytorch/audio upload-artifact: docs gpu-arch-type: cuda diff --git a/.github/workflows/unittest-linux-gpu.yml b/.github/workflows/unittest-linux-gpu.yml index f3c4da3eb5..19bb9cbd50 100644 --- a/.github/workflows/unittest-linux-gpu.yml +++ b/.github/workflows/unittest-linux-gpu.yml @@ -14,7 +14,7 @@ jobs: strategy: matrix: python-version: ["3.11"] - cuda-version: ["12.6"] + cuda-version: ["12.9"] ffmpeg-version: ["7"] fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main