fix

liuwenran · Nov 13, 2023 · 07170b7 · 07170b7
1 parent ba2ef2f
commit 07170b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml
@@ -121,7 +121,7 @@ jobs:
       - name: Install lmdb
         run: python -m pip install lmdb
       - name: Install PyTorch
-        run: python -m pip install torch==2.1.0+${{matrix.platform}} torchvision==0.16.0+${{matrix.platform}}
+        run: python -m pip install torch==2.1.0+${{matrix.platform}} torchvision==0.16.0+${{matrix.platform}} -f https://download.pytorch.org/whl/torch_stable.html
       - name: Install mmagic dependencies
         run: |
           python -m pip install git+https://github.com/open-mmlab/mmengine.git@main