Spaces:
Build error
Build error
Commit
·
6332e4c
1
Parent(s):
07d4177
Trying other jax
Browse files- requirements.txt +1 -2
requirements.txt
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
tensorflow
|
| 2 |
-
jaxlib
|
| 3 |
-
jax
|
| 4 |
nest-asyncio
|
| 5 |
pyfluidsynth==1.3.0
|
| 6 |
jax[cuda]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
|
|
|
| 1 |
tensorflow
|
| 2 |
+
jaxlib==0.3.15+cuda11.cudnn82
|
|
|
|
| 3 |
nest-asyncio
|
| 4 |
pyfluidsynth==1.3.0
|
| 5 |
jax[cuda]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|