install cuda version of jax and jaxlib
6c4ca29
Alex Lau
commited on
Spaces:
Runtime error
Runtime error
Commit History
revert to CPU setting
368e60c
Alex Lau
commited on
change jax release url link
3745caa
Alex Lau
commited on
revert to orig jax dependencies
48b917c
cpu jax installation
6d2eea0
change jax cuda version
2244280
revert to cpu mode with downsample=4
61a957f
set CUDA jax
279af7e
downgrade jaxlib
0c4c744
spec jax cpu install
96899f1
comment jax cuda
6dba29d
downgrade flax to 0.3.4
b04a705
revert to CPU and fix jax, jaxlib, flax ver
eaa31f6
Revert "revert to cpu config"
bf2d3d5
revert to gpu mode
40a686d
revert to cpu config
b30f1b8
Add JAX CUDA version in requirements.txt
fc5bccc
Fixed requirements.txt
4caa204
Sasikanth
commited on