Spaces:
Runtime error
Runtime error
cudnn 8.2.0
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -6,5 +6,5 @@ einops
|
|
6 |
-f https://download.pytorch.org/whl/cpu/torch
|
7 |
torch[cpu]
|
8 |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
9 |
-
jax[
|
10 |
flax
|
|
|
6 |
-f https://download.pytorch.org/whl/cpu/torch
|
7 |
torch[cpu]
|
8 |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
9 |
+
jax[cuda11_cudnn82] #jax[cuda11_cudnn86] #jax[cuda11_cudnn805]
|
10 |
flax
|