search

kalininalab / alphafold_non_docker

AlphaFold2 non-docker setup
325 stars 119 forks source link

Jax #29

Closed kdpan closed 2 years ago

kdpan commented 2 years ago

Is any Jax version compatible with CUDA 10.2?

sanjaysrikakulam commented 2 years ago

Hi @kdpan

Please check here https://github.com/google/jax/discussions/6236

kdpan commented 2 years ago

Hi@sanjaysrikakulam Thank you!!! However, that didn't work，the version of jax requires version 0.1.74, and the cuda10. 2. only Jax 0.1.71 can be installed at most.

sanjaysrikakulam commented 2 years ago

Hi @kdpan

In that case, is it possible to downgrade your JAX version? or upgrade the CUDA version?

hegelab commented 2 years ago

This may be good to change in the README.md:

pip install --upgrade jax jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html

pip install jax==0.2.20 jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html

sanjaysrikakulam commented 2 years ago

I have made necessary changes to the readme and I was unable to reproduce this in a new conda env with the latest changes. Hope it is fixed with the new changes.