From 9dac79b2097f82a686371fbac936e81ea2b572ed Mon Sep 17 00:00:00 2001
From: Vipul Ajmera <Vipul.Ajmera@ibm.com>
Date: Thu, 5 Feb 2026 21:33:18 +0530
Subject: [PATCH] Update torchvision_0.24.1_ubi_9.6.sh

---
 t/torchvision/torchvision_0.24.1_ubi_9.6.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/t/torchvision/torchvision_0.24.1_ubi_9.6.sh b/t/torchvision/torchvision_0.24.1_ubi_9.6.sh
index 5952dbd9ee..a19b24383f 100644
--- a/t/torchvision/torchvision_0.24.1_ubi_9.6.sh
+++ b/t/torchvision/torchvision_0.24.1_ubi_9.6.sh
@@ -591,6 +591,10 @@ cp libtorchvision.so $CURRENT_DIR/vision/torchvision/libtorchvision.so
 cp libtorchvision.so $PYTHON_SITE_PACKAGES/torch/share/cmake/Torch
 cp libtorchvision.so /usr/local/lib64
 cd $CURRENT_DIR/vision
+
+export LD_LIBRARY_PATH="${CURRENT_DIR}/pytorch/build/lib/:$LD_LIBRARY_PATH"
+export LD_LIBRARY_PATH="${CURRENT_DIR}/protobuf/local/libprotobuf/lib64/:$LD_LIBRARY_PATH"
+
 python3 setup.py bdist_wheel --dist-dir $CURRENT_DIR
 
 cd $CURRENT_DIR
@@ -610,4 +614,3 @@ else
     echo "$PACKAGE_NAME  |  $PACKAGE_URL | $PACKAGE_VERSION | $OS_NAME | GitHub  | Pass |  Both_Install_and_Test_Success"
     exit 0
 fi
-