fix wheels to manylinux

NVIDIA · Aug 27, 2024 · d4ecf1c · d4ecf1c
1 parent 222114f
commit d4ecf1c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/.github/workflows/deploy_pypi.yml b/.github/workflows/deploy_pypi.yml
@@ -28,12 +28,21 @@ jobs:
           python3 -m pip install numpy
           python3 -m pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
 
+      - name: Install auditwheel
+        run: |
+          python3 -m pip install auditwheel
+
       - name: Build distribution
         run: |
-          python3 -m build
+          python3 -m build --wheel --sdist
+
+      - name: Repair Linux wheel with auditwheel
+        run: |
+          auditwheel repair dist/*.whl -w wheelhouse/
 
       - name: Publish package to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           user: __token__
-          password: ${{ secrets.PYPI_PASSWORD }}
+          password: ${{ secrets.PYPI_PASSWORD }}
+          packages_dir: wheelhouse/