diff --git a/.github/workflows/pr-stage-check.yml b/.github/workflows/pr-stage-check.yml index 9d54247d..8fa3c1ff 100644 --- a/.github/workflows/pr-stage-check.yml +++ b/.github/workflows/pr-stage-check.yml @@ -106,7 +106,7 @@ jobs: - name: Upgrade pip run: python -m pip install pip --upgrade - name: Install PyTorch - run: pip install torch==${{matrix.torch}} -f https://download.pytorch.org/whl/${{matrix.platform}}/torch_stable.html + run: pip install torch==${{matrix.torch}} -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Install opencompass dependencies run: | pip install -r requirements.txt