set cuda to 12.1 & pytorch to nightly in ci

deftlion · Dec 6, 2024 · 1867eac · 1867eac
1 parent f398d43
commit 1867eac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -49,8 +49,8 @@ jobs:
       matrix:
           os: ['ubuntu-20.04']
           python-version: ['3.10']
-          pytorch-version: ['2.5.1']  # Must be the most recent version that meets requirements-cuda.txt.
-          cuda-version: ['12.2']
+          pytorch-version: ['nightly']  # Must be the most recent version that meets requirements-cuda.txt.
+          cuda-version: ['12.1']
 
     steps:
       - name: Checkout