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.

pip installing jaxlib 0.1.63 fails on Mac OS

See original GitHub issue

Upgrading from jaxlib 0.1.62 to 0.1.63 causes import of jax to fail with mysterious error messages. Downgrading back to 0.1.62 and everything is happy again. (This in on Mac OS)

Requirement already satisfied: jaxlib in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (0.1.62)
Collecting jaxlib
  Using cached jaxlib-0.1.63-cp38-none-macosx_10_9_x86_64.whl (42.7 MB)
Requirement already satisfied: absl-py in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (0.12.0)
Requirement already satisfied: numpy>=1.16 in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (1.20.1)
Requirement already satisfied: flatbuffers in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (20210203211748)
Requirement already satisfied: scipy in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (1.6.1)
Requirement already satisfied: six in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from absl-py->jaxlib) (1.15.0)
Installing collected packages: jaxlib
  Attempting uninstall: jaxlib
    Found existing installation: jaxlib 0.1.62
    Uninstalling jaxlib-0.1.62:
      Successfully uninstalled jaxlib-0.1.62
Successfully installed jaxlib-0.1.63

$ python -c "import jax"
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/__init__.py", line 21, in <module>
    from .config import config
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/config.py", line 19, in <module>
    from jax import lib
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/lib/__init__.py", line 53, in <module>
    from jaxlib import xla_client
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_client.py", line 31, in <module>
    from . import xla_extension as _xla
ImportError: dlopen(/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so, 2): Symbol not found: __ZN10tensorflow8profiler11MonitorGrpcERKNSt3__112basic_stringIcNS1_11char_traitsIcEENS1_9allocatorIcEEEERKNS_14MonitorRequestEPNS_15MonitorResponseE
  Referenced from: /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so
  Expected in: flat namespace
 in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so

Issue Analytics

State:
Created 3 years ago
Reactions:1
Comments:11 (1 by maintainers)

Top GitHub Comments

4reactions

skyecommented, Mar 18, 2021

Oof, after lovingly fixing this error and building new release wheels, I uploaded the old broken wheels! 🤦‍♀️ I don’t think pypi lets you upload new wheels for a given release, but I missed something else in the 0.1.63 release anyway so am gonna cut a 0.1.64 soon (like tomorrow). So I recommend sticking to 0.1.62 for now, and upgrading to 0.1.64 once it’s available (I can close this issue when that happens).

1reaction

skyecommented, Mar 24, 2021

Oops, forgot to close this issue, but jaxlib 0.1.64 has been available for about a week and shouldn’t have this issue!

Top Results From Across the Web

importing jax fails on mac with m1 chip - python - Stack Overflow

I tried using jax with pip install jax , then I built it from source as suggested by the comment, by cloning their...

jaxlib - PyPI

jaxlib is the support library for JAX. While JAX itself is a pure Python package, jaxlib contains the binary (C/C++) parts of the...

Change log - JAX documentation

Fixes an incompatibility between jax 0.3.19 and the libtpu version it pinned (#12550). Requires jaxlib 0.3.20. Fix incorrect pip url in setup.py comment ......

Jaxlib - :: Anaconda.org

osx -arm64 v0.4.1; linux-64 v0.4.1; osx-64 v0.4.1. conda install. To install this package run one of the following: conda install -c conda-forge jaxlib...

[chimerax-users] alphafold in Google colab - UCSF RBVI

kapre 0.3.7 requires tensorflow>=2.0.0, which is not installed. ... wrapt-1.12.1 > + pip3 install --upgrade jaxlib==0.3.2+cuda11.cudnn82 -f ...