fix

vectorch-ai · Oct 14, 2024 · 314b9ed · 314b9ed
1 parent 84877f0
commit 314b9ed
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/package_test.yml b/.github/workflows/package_test.yml
@@ -43,9 +43,13 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
+        python: ["3.10"]
+        cuda: ["12.4"]
         torch: ["2.4.1"]
     runs-on: [self-hosted, linux, build]
     env:
+      PYTHON_VERSION: ${{ matrix.python }}
+      CUDA_VERSION: ${{ matrix.cuda }}
       TORCH_VERSION: ${{ matrix.torch }}
 
     steps: