jax is already installed

QuantEcon · Jun 2, 2024 · 4d75f11 · 4d75f11
1 parent d72c0fe
commit 4d75f11
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -27,11 +27,11 @@ jobs:
         if: steps.cache-docker-mmcky.outputs.cache-hit == 'true'
         run: docker image load --input ./ci/cache/docker/cuda/docker-img.tar
 
-      - name: Install JAX[CUDA] and Numpyro[CUDA]
-        shell: bash -l {0}
-        run: |
-          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: Install JAX[CUDA] and Numpyro[CUDA]
+      #   shell: bash -l {0}
+      #   run: |
+      #     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
         shell: bash -l {0}