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

Signed-off-by: Ming M Qin <[email protected]>
Project-MONAI · Apr 17, 2024 · ea0c032 · ea0c032
1 parent 3e3404e
commit ea0c032
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/.github/workflows/pr.yml b/.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