install torch cpu only

MIT-SPARK · Jul 19, 2024 · 33d1757 · 33d1757
1 parent 5288d7e
commit 33d1757
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/python.yaml b/.github/workflows/python.yaml
@@ -26,5 +26,6 @@ jobs:
         run: python -m pip install ${{github.workspace}}
       - name: Test
         run: |
-          pip install pytest networkx torch torch_geometric
+          pip install torch --index-url https://dowload.pytorch.org/whl/cpu
+          pip install pytest networkx torch_geometric
           pytest ${{github.workspace}}/python/tests