diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index 17f18fe8b..a9911f234 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -140,6 +140,7 @@ jobs: ls - name: Run Python tests run: | + ${CONDA_RUN} export LD_LIBRARY_PATH=${CONDA_PREFIX}/lib/:/usr/local/cuda/lib64 ${CONDA_RUN} FAIL_WITHOUT_CUDA=1 pytest --override-ini="addopts=-v" test --tb=short - name: Run Python benchmark run: |