Can't install JAX with poetry

See original GitHub issue
  • I am on the latest Poetry version.
  • I have searched the issues of this repo and believe that this is not a duplicate.
  • If an exception occurs when executing a command, I executed it again in debug mode (-vvv option).

Issue

Hello,

I am trying to install JAX (v.0.3.6) using poetry by running poetry add jax, which unfortunately fails, and returns Failed to add packages, reverting the pyproject.toml file to its original content.. I can post the full error message, that is very long. However if I run pip install jax, it installs without any problem.

Has someone encountered the same issue and/or has an idea how to fix it ?

Thank you!

Issue Analytics

  • State:closed
  • Created a year ago
  • Comments:7 (2 by maintainers)

github_iconTop GitHub Comments

2reactions
sudomazecommented, May 24, 2022

I have tried this and it worked out!

jaxlib = { extras = ["tpu"], version = "0.3.10" }

I have run a simple unit test to check if tpu was enabled and it seems. Thanks!

1reaction
dimblebycommented, Apr 23, 2022

good to close, then?

Read more comments on GitHub >

github_iconTop Results From Across the Web

Not able to install jaxlib - Stack Overflow
create a virtual python environment with the packages pip, setuptools, six, wheel and numpy (i did it in the GUI of pycharm). download...
Read more >
Building from source - JAX documentation
Building JAX involves two steps: Building or installing jaxlib , the C++ support library for jax . Installing the jax Python package.
Read more >
Announcing Poetry 1.2.0 | Blog
Poetry will be installed using the Python interpreter the installer is invoked with. The new installer (or a compatible install method such as ......
Read more >
google/jax を Poetry でインストールする - Zenn
Can't install JAX with poetry · Issue #5481 · python-poetry/poetry ... poetry add --source jax jaxlib $ poetry add jax.
Read more >
jaxlib - PyPI
XLA library for JAX. ... pip install jaxlib ... While JAX itself is a pure Python package, jaxlib contains the binary (C/C++) parts...
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found