diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml index 1d237f5d8ef..2d30771e75a 100644 --- a/.github/workflows/cuda.yml +++ b/.github/workflows/cuda.yml @@ -24,7 +24,7 @@ jobs: strategy: fail-fast: false matrix: - cuda-version: ["12.6", "12.8", "13.0"] + cuda-version: ["12.6", "12.8", "12.9", "13.0"] name: test-executorch-cuda-build-${{ matrix.cuda-version }} uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main diff --git a/install_requirements.py b/install_requirements.py index 4a8d9ed8870..df26badbef3 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -23,6 +23,7 @@ SUPPORTED_CUDA_VERSIONS = ( (12, 6), (12, 8), + (12, 9), (13, 0), )