[ci] Update CUDA versions for CI (#6539)

* Update CUDA versions for CI * use latest CUDA --------- Co-authored-by: James Lamb <[email protected]>
microsoft · Oct 6, 2024 · 718da7d · 718da7d
1 parent 436f826
commit 718da7d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml
@@ -82,13 +82,13 @@ jobs:
           - method: wheel
             compiler: gcc
             python_version: "3.10"
-            cuda_version: "11.8.0"
-            linux_version: "ubuntu20.04"
+            cuda_version: "12.6.1"
+            linux_version: "ubuntu22.04"
             task: cuda
           - method: source
             compiler: gcc
             python_version: "3.12"
-            cuda_version: "12.2.0"
+            cuda_version: "12.2.2"
             linux_version: "ubuntu22.04"
             task: cuda
           - method: pip