score_inverse_problems icon indicating copy to clipboard operation
score_inverse_problems copied to clipboard

Published 20 hours ago • verified publisher icon yang-song

Request for updating the dependency

Open GreameLee opened this issue 1 year ago • 1 comments

The dependency is really hard to manage the dependency version to install all of them. Especially flax @ git+https://github.com/google/flax.git@8b2e56e93602ff9fcd62647f28b8145e0e0c10a8, I would like say that there is no way to install all required pakages now

GreameLee avatar Sep 23 '23 05:09 GreameLee

Hi @GreameLee , I agree with you,took me forever to get this dependency working. Here are the steps

  • create environment with python 3.9
  • conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
  • change following in requirements.txt(keep others same) : jaxlib==0.1.69 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html torch==1.10.1

yeswanthreddyguddeti avatar Feb 16 '24 04:02 yeswanthreddyguddeti