Skip to content

Commit cdacbe0

Browse files
committed
Fix nvjpeg packaging into the wheel (#4752)
Relocate fails to find nvjpeg unless its present in the library search path.
1 parent fa347eb commit cdacbe0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

packaging/build_wheel.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,6 @@ else
5555
if [[ "$OSTYPE" == "msys" ]]; then
5656
"$script_dir/windows/internal/vc_env_helper.bat" python $script_dir/wheel/relocate.py
5757
else
58-
LD_LIBRARY_PATH="/usr/local/lib:$LD_LIBRARY_PATH" python $script_dir/wheel/relocate.py
58+
LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python $script_dir/wheel/relocate.py
5959
fi
6060
fi

0 commit comments

Comments
 (0)