Skip to content

Commit

Permalink
remove numpyro
Browse files Browse the repository at this point in the history
  • Loading branch information
mmcky committed May 13, 2024
1 parent a1d4efb commit 11fc84f
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ jobs:
- name: Install JAX[CUDA] and Numpyro[CUDA]
shell: bash -l {0}
run: |
pip install numpyro[cuda] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
pip install --upgrade "jax[cuda12_local]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
- name: Check nvidia drivers
shell: bash -l {0}
Expand Down

0 comments on commit 11fc84f

Please sign in to comment.