Update CUDA version on GPU tests (#2040)

PyTorch core is dropping support for 11.6.
pytorch · Feb 2, 2023 · 04a4c30 · 04a4c30
1 parent 8a077d8
commit 04a4c30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: