search

google-deepmind / alphageometry

Apache License 2.0
4.19k stars 469 forks source link

Issue with version of flax and jax while installing in WSL #121

Closed Nishchal7021 closed 5 months ago

Nishchal7021 commented 5 months ago

The problem is env requires latest version of jax i.e 0.4.28 , but it does not have linear_util , which is fine until you have latest version of flax i.e 0.8.4 , but latest version of flax does not have optim , and to have optim you need to downgrade flax to 0.5.3 , but if you do that you will need linear_util, so im stuck in this loop , what to do? image image image

Nishchal7021 commented 5 months ago

I might've forgotten to activate the virtual environment before installing dependencies , I did it again and it worked without having to install any dependencies.