From 9aa93a393224db4979b925f26ef36c637d0ffa94 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 25 Apr 2024 21:02:20 +0800 Subject: [PATCH] fix cuda --- scripts/github_actions/build-ubuntu-cuda.sh | 24 +++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) 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