lopho commited on
Commit
5c844a3
1 Parent(s): 4b1a319

cudnn 8.2.0

Browse files
Files changed (1) hide show
  1. 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[cuda11_cudnn805] #jax[cuda11_cudnn86] #jax[cuda11_cudnn805]
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