DietNerf-Demo / requirements.txt
hassiahk's picture
Add JAX CUDA version in requirements.txt
fc5bccc
raw
history blame
No virus
319 Bytes
--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
jax[cuda111]
numpy>=1.16.4
# jax>=0.2.6
# jaxlib>=0.1.57
flax>=0.2.2
# opencv-python>=4.4.0
Pillow>=7.2.0
pyyaml>=5.3.1
tensorboard>=2.4.0
tensorflow>=2.3.1
tensorflow-hub>=0.11.0
transformers==4.8.2
tqdm==4.61.2
streamlit==0.84.1
gdown==3.13.0