Spaces:
Build error
Build error
Add JAX CUDA version in requirements.txt
Browse files- requirements.txt +5 -4
requirements.txt
CHANGED
|
@@ -1,9 +1,10 @@
|
|
|
|
|
|
|
|
| 1 |
numpy>=1.16.4
|
| 2 |
-
jax>=0.2.6
|
| 3 |
-
#
|
| 4 |
-
jaxlib>=0.1.57
|
| 5 |
flax>=0.2.2
|
| 6 |
-
#opencv-python>=4.4.0
|
| 7 |
Pillow>=7.2.0
|
| 8 |
pyyaml>=5.3.1
|
| 9 |
tensorboard>=2.4.0
|
|
|
|
| 1 |
+
--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
|
| 2 |
+
jax[cuda111]
|
| 3 |
numpy>=1.16.4
|
| 4 |
+
# jax>=0.2.6
|
| 5 |
+
# jaxlib>=0.1.57
|
|
|
|
| 6 |
flax>=0.2.2
|
| 7 |
+
# opencv-python>=4.4.0
|
| 8 |
Pillow>=7.2.0
|
| 9 |
pyyaml>=5.3.1
|
| 10 |
tensorboard>=2.4.0
|