Update pr.yml

Repeat setting lib path for cuda12 because it is not found in Unit Tests Signed-off-by: Ming M Qin <[email protected]>
Project-MONAI · Apr 10, 2024 · a83f932 · a83f932
1 parent a429e53
commit a83f932
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/pr.yml b/.github/workflows/pr.yml
@@ -39,11 +39,13 @@ jobs:
         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
           ./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
           coverage xml
       - name: Upload coverage
         uses: codecov/codecov-action@v2