diff --git a/.github/workflows/publish-to-pypi.yml b/.github/workflows/publish-to-pypi.yml
index b9c6de2a2e..5ed8e7cd96 100644
--- a/.github/workflows/publish-to-pypi.yml
+++ b/.github/workflows/publish-to-pypi.yml
@@ -17,7 +17,7 @@ jobs:
         with:
           python-version: 3.7
       - name: Upgrade Setuptools
-        run: pip install setuptools --upgrade
+        run: pip install setuptools wheel --upgrade
       - name: Build MMCV
         run: |
           sed -i "s/os.getenv('MMCV_WITH_OPS', '1')/os.getenv('MMCV_WITH_OPS', '0')/g" setup.py