From 804afd42d36bca2072c8056a7a1e3fe6d6d9f788 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Thu, 27 Jun 2024 12:48:52 -0700 Subject: [PATCH] Exclude driver stub. --- ci/build_wheel.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh index 3c5cc86..31d0017 100755 --- a/ci/build_wheel.sh +++ b/ci/build_wheel.sh @@ -12,7 +12,8 @@ rapids-logger "Build wheel" mkdir -p ./dist python -m pip wheel . --wheel-dir=./dist -vvv --disable-pip-version-check --no-deps -python -m auditwheel repair -w ./final_dist ./dist/* +# Exclude libcuda.so.1 because we only install a driver stub +python -m auditwheel repair --exclude libcuda.so.1 -w ./final_dist ./dist/* rapids-logger "Upload Wheel" RAPIDS_PY_WHEEL_NAME="pynvjitlink_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 ./final_dist