Pin nvidia-cudnn-cu12==8.9.7.29

cudnn 9 is not uploaded to pypi due to pypi restrictions
google · Mar 27, 2024 · e9fc40f · e9fc40f
1 parent 4e4d7f5
commit e9fc40f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+nvidia-cudnn-cu12==8.9.7.29
 jax>=0.4.23
 jaxlib>=0.4.23
 orbax-checkpoint>=0.5.5