diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh index 08b282e62d..976da98998 100755 --- a/ci/build_wheel.sh +++ b/ci/build_wheel.sh @@ -22,16 +22,13 @@ rapids-generate-version > ./VERSION cd "${package_dir}" EXCLUDE_ARGS=( - --exclude "libcublas.so.11" - --exclude "libcublas.so.12" - --exclude "libcublasLt.so.11" - --exclude "libcublasLt.so.12" - --exclude "libcurand.so.10" - --exclude "libcusolver.so.11" - --exclude "libcusparse.so.11" - --exclude "libcusparse.so.12" - --exclude "libnvJitLink.so.12" - --exclude "libucp.so.0" + --exclude "libcublas.so.*" + --exclude "libcublasLt.so.*" + --exclude "libcurand.so.*" + --exclude "libcusolver.so.*" + --exclude "libcusparse.so.*" + --exclude "libnvJitLink.so.*" + --exclude "libucp.so.*" ) if [[ ${package_name} != "libraft" ]]; then diff --git a/python/libraft/CMakeLists.txt b/python/libraft/CMakeLists.txt index 541e07b23d..db81aa9507 100644 --- a/python/libraft/CMakeLists.txt +++ b/python/libraft/CMakeLists.txt @@ -48,9 +48,9 @@ add_subdirectory(../../cpp raft-cpp) set_property( TARGET raft_lib PROPERTY INSTALL_RPATH - "$ORIGIN/../../nvidia/cublas/lib" - "$ORIGIN/../../nvidia/curand/lib" - "$ORIGIN/../../nvidia/cusolver/lib" - "$ORIGIN/../../nvidia/cusparse/lib" - "$ORIGIN/../../nvidia/nvjitlink/lib" + "$ORIGIN/../../nvidia/cublas/lib" + "$ORIGIN/../../nvidia/curand/lib" + "$ORIGIN/../../nvidia/cusolver/lib" + "$ORIGIN/../../nvidia/cusparse/lib" + "$ORIGIN/../../nvidia/nvjitlink/lib" )