Windows GPU unittest: try 3.10 and cuda 12.1

pytorch · Aug 11, 2024 · 89647a1 · 89647a1
1 parent 8dc06a6
commit 89647a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -79,10 +79,10 @@ jobs:
         runner: ["windows.4xlarge"]
         gpu-arch-type: ["cpu"]
         include:
-          - python-version: "3.9"
+          - python-version: "3.10"
             runner: windows.g5.4xlarge.nvidia.gpu
             gpu-arch-type: cuda
-            gpu-arch-version: "11.8"
+            gpu-arch-version: "12.1"
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
     with: