update publish-pip

xinntao · May 15, 2021 · 696ecaf · 696ecaf
1 parent d0427c0
commit 696ecaf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/publish-pip.yml b/.github/workflows/publish-pip.yml
@@ -14,7 +14,10 @@ jobs:
         with:
           python-version: 3.8
       - name: Upgrade pip
-        run: pip install pip --upgrade
+        run: |
+          pip install pip --upgrade
+          pip install wheel
+
       - name: Install PyTorch (cpu)
         run: pip install torch==1.7.0+cpu torchvision==0.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
       - name: Install dependencies