diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index bd5b46e..771185c 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -58,7 +58,7 @@ jobs: - name: Install build tools run: | - python -m pip install --upgrade pip build wheel + python -m pip install --upgrade pip auditwheel build wheel sudo apt-get update sudo apt-get install -y --no-install-recommends cmake ninja-build patchelf wget ca-certificates @@ -92,7 +92,14 @@ jobs: --cmake-arg="-DCMAKE_CUDA_ARCHITECTURES=$CUDA_ARCHITECTURES" python -m build --wheel mkdir -p wheelhouse - cp dist/*.whl wheelhouse/ + python -m auditwheel repair \ + --plat manylinux_2_39_${{ matrix.arch }} \ + --exclude libcudart.so.12 \ + --exclude libcublas.so.12 \ + --exclude libcublasLt.so.12 \ + --exclude libcuda.so.1 \ + -w wheelhouse \ + dist/*.whl - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1 with: diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index 6137e5c..5e3a356 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -82,7 +82,7 @@ jobs: - name: Install build tools run: | - python -m pip install --upgrade pip build wheel + python -m pip install --upgrade pip auditwheel build wheel sudo apt-get update sudo apt-get install -y --no-install-recommends cmake ninja-build patchelf wget ca-certificates @@ -116,7 +116,17 @@ jobs: --cmake-arg="-DCMAKE_CUDA_ARCHITECTURES=$CUDA_ARCHITECTURES" python -m build --wheel mkdir -p wheelhouse - cp dist/*.whl wheelhouse/ + python -m auditwheel repair \ + --plat manylinux_2_39_${{ matrix.arch }} \ + --exclude libcudart.so.12 \ + --exclude libcudart.so.13 \ + --exclude libcublas.so.12 \ + --exclude libcublas.so.13 \ + --exclude libcublasLt.so.12 \ + --exclude libcublasLt.so.13 \ + --exclude libcuda.so.1 \ + -w wheelhouse \ + dist/*.whl - uses: actions/upload-artifact@v4 with: @@ -148,7 +158,7 @@ jobs: - name: Install build tools run: | - python -m pip install --upgrade pip build wheel + python -m pip install --upgrade pip auditwheel build wheel sudo apt-get update sudo apt-get install -y --no-install-recommends cmake ninja-build patchelf ca-certificates @@ -168,7 +178,10 @@ jobs: --cmake-arg=Ninja python -m build --wheel mkdir -p wheelhouse - cp dist/*.whl wheelhouse/ + python -m auditwheel repair \ + --plat manylinux_2_39_${{ matrix.arch }} \ + -w wheelhouse \ + dist/*.whl - uses: actions/upload-artifact@v4 with: