search

ikostrikov / implicit_q_learning

MIT License
226 stars 38 forks source link

conflicting dependencies between optax and jaxlib #3

Closed enosair closed 2 years ago

enosair commented 2 years ago

Hi, I got the following error when running pip install -r ./requirements.txt

The conflict is caused by: optax 0.0.9 depends on jaxlib>=0.1.37 optax 0.0.8 depends on jaxlib>=0.1.37 optax 0.0.6 depends on jaxlib>=0.1.37

Could you please take a look? Thank you.

ikostrikov commented 2 years ago

I cannot reproduce the issue on Ubuntu 20.04 and python 3.8.

What OS and python version are you using?

enosair commented 2 years ago

Hi, thanks for the prompt reply. I'm using Ubuntu 20.04.1 and Python 3.10.0.

On Sun, Oct 31, 2021 at 12:10 AM Ilya Kostrikov @.***> wrote:

I cannot reproduce the issue on Ubuntu 20.04 and python 3.8.

What OS and python version are you using?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/ikostrikov/implicit_q_learning/issues/3#issuecomment-955633550, or unsubscribe https://github.com/notifications/unsubscribe-auth/AA4SJ5QHIUQCYB2C7IE2B53UJS6VDANCNFSM5HBYCCFQ . Triage notifications on the go with GitHub Mobile for iOS https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675 or Android https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub.

ikostrikov commented 2 years ago

It seems that jaxlib does not support python 3.10 yet: https://github.com/google/jax/issues/8279

See available releases at https://storage.googleapis.com/jax-releases/jax_releases.html

enosair commented 2 years ago

It seems that jaxlib does not support python 3.10 yet: google/jax#8279

See available releases at https://storage.googleapis.com/jax-releases/jax_releases.html

Got it. I downgraded to python 3.8 and can solve this issue. However I hit another problem during installing gym[mujoco] -- it was looking for mujoco150 but I have mujoco200.

I ignored that line and manually installed the other packages, now I can run your code.

enosair commented 2 years ago

When I try to install jaxlib with cuda support, it automatically upgrade my jax to 0.2.24. Is it possible to use jax 0.2.21 (as specified in your requirements.txt) with GPU?

ikostrikov commented 2 years ago

I've added cuda instructions to the readme. I'm sorry for the confusion. Also, the instructions in jaxrl were incorrect. I fixed them there as well.