From f29ec8e6ee36d1fef78a9d9bf3ee6448f44f8133 Mon Sep 17 00:00:00 2001
From: Nikita Shulga <nshulga@fb.com>
Date: Tue, 26 Oct 2021 08:04:27 -0700
Subject: [PATCH] Fix nvjpeg packaging into the wheel

Relocate fails to find nvjpeg unless its present in the library search path.
---
 packaging/build_wheel.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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