compile torvhsion from source

pytorch · Mar 7, 2024 · bcdc721 · bcdc721
1 parent 62399f1
commit bcdc721
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/.github/workflows/tpu_ci.yml b/.github/workflows/tpu_ci.yml
@@ -26,14 +26,17 @@ jobs:
         run: |
           cd pytorch/xla
           python3 setup.py install --user
+      - name: Install torchvision
+        run: |
+          cd pytorch/
+          git clone https://github.com/pytorch/vision.git
+          cd vision
+          python3 setup.py install --user
       - name: Run Tests
         env:
           PJRT_DEVICE: TPU
         run: |
           pip install fsspec
           pip install rich
-          pip uninstall torch
-          pip uninstall torchvision
-          pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
           cd pytorch/xla
           test/tpu/run_tests.sh