Runtime Error when trying to start plugin on M2 Chip

[jenna-tomkinson]

Problem

I was excited to look into the BaSiCPy plugin for napari but I have run into two different errors that states:

RuntimeError: Failed to import command at 'napari_basicpy._widget:BasicWidget': partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import)

RuntimeError: Failed to import command at 'napari_basicpy._widget:BasicWidget': This version of jaxlib was built using AVX instructions, which your CPU and/or operating system do not support. You may be able work around this issue by building jaxlib from source.

It looks like the jaxlib package is the issue. When looking at this repo, I do not see a solution for this. But, there is a solution on the Python package version of BaSiCPy's Github: https://github.com/peng-lab/BaSiCPy.

Solution

It would also be helpful to include this solution (even though it is specific to M1 chips) on this repo.

pip install "jax[cpu]==0.3.22" jaxlib

But, this might not be the correct solution when using napari or for the M2 chip. I would love to know if this above solution referenced in the other repo also applies here.

[yfukai]

Thanks for this report @jenna-tomkinson! I found

pip install "jax[cpu]==0.3.22" jaxlib

actually does not help solve the problem (in M1 mac) but we need to install jax and jaxlib from conda-forge.

conda install -c jax jaxlib

We'll update the documents!