Update test workflow

Author: matthewdouglas

.github/workflows/tests.yml

@@ -49,7 +49,8 @@ jobs:
   build-cuda:
     strategy:
       matrix:
-        cuda_version: ["11.8.0", "12.6.3", "12.8.1", "12.9.1", "13.0.1"]
+        # TODO: Add 13.0.1 when we have runners with new enough drivers.
+        cuda_version: ["11.8.0", "12.6.3", "12.8.1", "12.9.1"]
         os: [ubuntu-22.04, ubuntu-22.04-arm]
         include:
           - os: ubuntu-22.04
@@ -111,7 +112,7 @@ jobs:
             arch: aarch64
           - os: ubuntu-22.04-arm
             arch: aarch64
-            torch_version: "2.5.1"
+            torch_version: "2.5.1"    # Higher minimum requirement for aarch64
           - os: windows-2025
             arch: x86_64
           - os: macos-15
@@ -182,7 +183,7 @@ jobs:
       - name: Setup Python
         uses: actions/setup-python@v5
         with:
-          python-version: 3.9
+          python-version: '3.10'
 
       - name: Install dependencies
         run: |
@@ -313,7 +314,7 @@ jobs:
       - name: Setup Python
         uses: actions/setup-python@v5
         with:
-          python-version: 3.9
+          python-version: '3.10'
 
       - name: Install PyTorch
         run: pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/xpu
@@ -343,23 +344,26 @@ jobs:
         os: [ubuntu-22.04, windows-2025]
         arch: [x86_64]
         gpu: [T4, L40S]
-        cuda_version: ["11.8.0", "12.6.3", "12.8.1", "12.9.1", "13.0.1"]
+        cuda_version: ["11.8.0", "12.6.3", "12.8.1", "12.9.1"] #, "13.0.1"]
         include:
           - cuda_version: "11.8.0"
             torch_version: "2.3.1"
             pypi_index: "https://download.pytorch.org/whl/cu118"
           - cuda_version: "12.6.3"
             torch_version: "2.6.0"
             pypi_index: "https://download.pytorch.org/whl/cu126"
-          - cuda_version: "12.8.1"
-            torch_version: "2.7.1"
-            pypi_index: "https://download.pytorch.org/whl/cu128"
           - cuda_version: "12.9.1"
             torch_version: "2.8.0"
             pypi_index: "https://download.pytorch.org/whl/cu129"
-          - cuda_version: "13.0.1"
+          - cuda_version: "12.8.1"
             torch_version: "2.9.0"
-            pypi_index: "https://download.pytorch.org/whl/test/cu130"
+            pypi_index: "https://download.pytorch.org/whl/test/cu128"
+
+          # Note: Currently our runners do not have new enough drivers for CUDA 13.
+          # Add this when supported.
+          # - cuda_version: "13.0.1"
+          #   torch_version: "2.9.0"
+          #   pypi_index: "https://download.pytorch.org/whl/test/cu130"
 
 
           # Linux L40S runners