Bumps cuda11_cudnn82 to cuda11_cudnn86.

The newest `0.4.23` that is selected with `>=0.4.2` does not provide a cuda11_cudnn82 anymore: https://storage.googleapis.com/jax-releases/jax_cuda_releases.html See also #295 PiperOrigin-RevId: 604573717
google-research · Feb 6, 2024 · 927f7a7 · 927f7a7
1 parent 2ecbc73
commit 927f7a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vit_jax/requirements.txt b/vit_jax/requirements.txt
@@ -6,7 +6,7 @@ einops>=0.3.0
 flax>=0.6.4
 git+https://github.com/google/flaxformer
 
-jax[cuda11_cudnn82]>=0.4.2
+jax[cuda11_cudnn86]>=0.4.2
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
 ml-collections>=0.1.0