Revert "try cuda_pip installer"

This reverts commit 6b3a167.
QuantEcon · May 10, 2024 · a1d4efb · a1d4efb
1 parent 6b3a167
commit a1d4efb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -20,7 +20,7 @@ jobs:
         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_pip]" -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}
         run: |