Request for updating the dependency

yang-song / score_inverse_problems

Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"

214 stars 26 forks source link

Open GreameLee opened 1 year ago

GreameLee commented 1 year ago

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

yeswanthreddyguddeti commented 7 months ago

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