From 11fc84fa0ebc4871d0d4f5c518daf7e453424c6b Mon Sep 17 00:00:00 2001 From: mmcky Date: Mon, 13 May 2024 15:20:02 +1000 Subject: [PATCH] remove numpyro --- .github/workflows/ci.yml | 1 - 1 file changed, 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 79de096..ba93569 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -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}