We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b00b3cc commit 0b09e45Copy full SHA for 0b09e45
requirements-jax-cuda.txt
@@ -10,6 +10,6 @@ torchvision>=0.16.0
10
# Jax with cuda support.
11
# Keep same version as Keras repo.
12
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
13
-jax[cuda12_pip]==0.6.0
+jax[cuda12]==0.6.0
14
15
-r requirements-common.txt
0 commit comments