Update pr.yml to set LD_LIBRARY_PATH consistently with Holoscan SDK 2.0

Signed-off-by: Ming M Qin <38891913+MMelQin@users.noreply.github.com>

Author: MMelQin

.github/workflows/pr.yml

@@ -32,20 +32,23 @@ jobs:
           python3 -c 'import sys; print(sys.executable)'
           python3 -c 'import site; print(site.getsitepackages())'
           python3 -m pip freeze
-          export LD_LIBRARY_PATH=`pwd`/.venv/lib/python3.8/site-packages/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH
+          export CUDA_WHL_LIB_DIR=$(python3 -c 'import nvidia.cuda_runtime; print(nvidia.cuda_runtime.__path__[0])')/lib
+          export LD_LIBRARY_PATH="$CUDA_WHL_LIB_DIR:$LD_LIBRARY_PATH"
           python3 -c 'from holoscan.core import *'
           ./run check -f
       - name: Run Unit tests
         run: |
           source .venv/bin/activate
           python3 -m pip install nvidia-cuda-runtime-cu12
-          export LD_LIBRARY_PATH=`pwd`/.venv/lib/python3.8/site-packages/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH
+          export CUDA_WHL_LIB_DIR=$(python3 -c 'import nvidia.cuda_runtime; print(nvidia.cuda_runtime.__path__[0])')/lib
+          export LD_LIBRARY_PATH="$CUDA_WHL_LIB_DIR:$LD_LIBRARY_PATH"
           ./run test all unit
       - name: Coverage
         run: |
           source .venv/bin/activate
           python3 -m pip install nvidia-cuda-runtime-cu12
-          export LD_LIBRARY_PATH=`pwd`/.venv/lib/python3.8/site-packages/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH
+          export CUDA_WHL_LIB_DIR=$(python3 -c 'import nvidia.cuda_runtime; print(nvidia.cuda_runtime.__path__[0])')/lib
+          export LD_LIBRARY_PATH="$CUDA_WHL_LIB_DIR:$LD_LIBRARY_PATH"
           coverage xml
       - name: Upload coverage
         uses: codecov/codecov-action@v2