diff --git a/packaging/build_wheel.sh b/packaging/build_wheel.sh index eb5e09e9dfd..51ebe95d987 100755 --- a/packaging/build_wheel.sh +++ b/packaging/build_wheel.sh @@ -55,6 +55,6 @@ else if [[ "$OSTYPE" == "msys" ]]; then "$script_dir/windows/internal/vc_env_helper.bat" python $script_dir/wheel/relocate.py else - LD_LIBRARY_PATH="/usr/local/lib:$LD_LIBRARY_PATH" python $script_dir/wheel/relocate.py + LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python $script_dir/wheel/relocate.py fi fi