diff --git a/scripts/github_actions/build-ubuntu-cuda.sh b/scripts/github_actions/build-ubuntu-cuda.sh index c3d1423..ad1dc00 100755 --- a/scripts/github_actions/build-ubuntu-cuda.sh +++ b/scripts/github_actions/build-ubuntu-cuda.sh @@ -96,13 +96,33 @@ auditwheel --verbose repair \ --exclude libtorch_cuda.so \ --exclude libtorch_python.so \ \ - --exclude libcudnn.so.8 \ + --exclude libcublas.so \ --exclude libcublas.so.11 \ + --exclude libcublas.so.12 \ + --exclude libcublasLt.so \ --exclude libcublasLt.so.11 \ + --exclude libcublasLt.so.12 \ --exclude libcudart.so.11.0 \ + --exclude libcudart.so.12 \ + --exclude libcudnn.so.8 \ + --exclude libcufft.so \ + --exclude libcufft.so.11 \ + --exclude libcupti.so \ + --exclude libcupti.so.12 \ + --exclude libcurand.so \ + --exclude libcurand.so.10 \ + --exclude libcusparse.so \ + --exclude libcusparse.so.12 \ + --exclude libnccl.so \ + --exclude libnccl.so.2 \ + --exclude libnvJitLink.so \ + --exclude libnvJitLink.so.12 \ + --exclude libnvrtc.so \ --exclude libnvrtc.so.11.2 \ - --exclude libtorch_cuda_cu.so \ + --exclude libnvrtc.so.12 \ + --exclude libshm.so \ --exclude libtorch_cuda_cpp.so \ + --exclude libtorch_cuda_cu.so \ --plat manylinux_2_17_x86_64 \ -w /var/www/wheelhouse \ dist/*.whl