From 4d75f115011c7e28dc46e2db13116dc02a44c201 Mon Sep 17 00:00:00 2001 From: kp992 Date: Sun, 2 Jun 2024 22:49:51 +0530 Subject: [PATCH] jax is already installed --- .github/workflows/ci.yml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 9a0bcd1..a9cac2f 100644 --- 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}