RuntimeError (This version of jaxlib was built using AVX instructions) on importing `optimize_media` from `lightweight_mmm` in a conda environment on M1 mac.

[SergeyBondarev]

Description:

After setting up a new conda environment and installing the lightweight_mmm package, I encountered a RuntimeError when attempting to import the optimize_media function.

Steps to Reproduce:

• Create a new conda environment: conda create --name test_env python=3.10
• Activate the environment: conda activate test_env
• Install the package: pip install lightweight_mmm
• Attempt to import the function: from lightweight_mmm import optimize_media

Expected Behavior: The optimize_media function should be imported without any errors.

Actual Behavior: I received the following error:

RuntimeError: 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.

Environment:

Operating System: macOS Monterey, version 12.2 Python Version: 3.10 (using conda) lightweight_mmm Version: 0.1.9

Possible Solution: I thought that building jaxlib from source might work around the issue, as suggested by the error message. But I wasn't able to build it on my machine. Solutions with isolation of environment in docker also didn't work.

Is there any idea how can I fix this error?

[80boys]

pip uninstall jax jaxlib conda install -c conda-forge jaxlib conda install -c conda-forge jax