install cpu jaxlib

pytorch · Apr 9, 2024 · e7ca0a2 · e7ca0a2
1 parent de25dd4
commit e7ca0a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/torch_xla2.yml b/.github/workflows/torch_xla2.yml
@@ -35,7 +35,7 @@ jobs:
         working-directory: experimental/torch_xla2
         run: |
           pip install -r test-requirements.txt
-          pip install -e .
+          pip install -e .[cpu]
       - name: Run tests
         working-directory: experimental/torch_xla2
         shell: bash