diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 3b260ef..bf16cbf 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -26,6 +26,7 @@ jobs: - name: Install JAX[CUDA] and Numpyro[CUDA] shell: bash -l {0} run: | + export LD_LIBRARY_PATH=$CONDA_PREFIX/lib/ pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html pip install --upgrade "numpyro[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html - name: Check nvidia drivers