alexlau commited on
Commit
96899f1
1 Parent(s): 6dba29d

spec jax cpu install

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,7 +1,7 @@
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
6
  flax==0.3.4
7
  # opencv-python>=4.4.0
 
1
  #--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
2
  #jax[cuda111]
3
  numpy>=1.16.4
4
+ jax[cpu]==0.2.17
5
  jaxlib==0.1.69
6
  flax==0.3.4
7
  # opencv-python>=4.4.0