diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml index 9b017a9811..bc9fef126b 100644 --- a/.github/workflows/test-linux-gpu.yml +++ b/.github/workflows/test-linux-gpu.yml @@ -17,7 +17,7 @@ jobs: strategy: matrix: python_version: ["3.8"] - cuda_arch_version: ["11.6"] + cuda_arch_version: ["11.7"] fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job.yml@main with: