From a1d4efbb5c98035f439918baa7bf565149145a58 Mon Sep 17 00:00:00 2001 From: mmcky Date: Fri, 10 May 2024 12:40:16 +1000 Subject: [PATCH] Revert "try cuda_pip installer" This reverts commit 6b3a1675b16a66982d3e1b7a50b7ba6118e6667b. --- .github/workflows/ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index a332e72..79de096 100644 --- 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: |