pypi_distribution.sh 411 Bytes
Newer Older
1
set -eux
2

3
rm -rf dist raw_dist
4

5
6
7
8
9
10
11
12
13
14
15
16
17
18
python -mpip install -U pip
python -mpip install -U build wheel auditwheel patchelf

export NO_VERSION_LABEL=1

python -m build --sdist -o dist
python -m build --wheel -o raw_dist

auditwheel repair -L /lib -w dist \
    --exclude libcuda.so.1 --exclude /usr/local/cuda\* --exclude /opt/amdgpu\* \
    --exclude /opt/rocm\* \
    raw_dist/*.whl

echo "Wheel built successfully."