diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml index d9b330ca99cec29f937a67a4e1b97f97fde05cc0..dcd8f902e333f2974ac612e115fd058d2813ff49 100644 --- a/.github/workflows/merge_stage_test.yml +++ b/.github/workflows/merge_stage_test.yml @@ -86,6 +86,8 @@ jobs: python-version: ${{ matrix.python-version }} - name: Upgrade pip run: python -m pip install pip --upgrade + - name: Upgrade wheel + run: python -m pip install wheel --upgrade - name: Install PyTorch run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Build MMEngine from source @@ -239,6 +241,8 @@ jobs: python-version: ${{ matrix.python-version }} - name: Upgrade pip run: pip install pip --upgrade + - name: Upgrade wheel + run: pip install wheel --upgrade - name: Install PyTorch run: pip install torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }} - name: Build MMEngine from source diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml index d11feda58a580f47bf28a2979f970a4f18017632..320262d47cbdfc2c1fe3ab79ef68e2774f890da0 100644 --- a/.github/workflows/pr_stage_test.yml +++ b/.github/workflows/pr_stage_test.yml @@ -34,6 +34,8 @@ jobs: python-version: ${{ matrix.python-version }} - name: Upgrade pip run: python -m pip install pip --upgrade + - name: Upgrade wheel + run: python -m pip install wheel --upgrade - name: Install PyTorch run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Build MMEngine from source