Spaces:
Build error
Build error
comment jax cuda
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
|
2 |
-
jax[cuda111]
|
3 |
numpy>=1.16.4
|
4 |
jax==0.2.17
|
5 |
jaxlib==0.1.69
|
|
|
1 |
+
#--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
|
2 |
+
#jax[cuda111]
|
3 |
numpy>=1.16.4
|
4 |
jax==0.2.17
|
5 |
jaxlib==0.1.69
|