pypi_distribution.sh 402 Bytes
Newer Older
root's avatar
init  
root committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
set -eux

rm -rf dist

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."