question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

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