搭载M1芯片的Mac设备导入jax库运行报错问题求助

核心原因：你当前安装的jaxlib是x86架构版本，依赖AVX指令集，而ARM架构的M1芯片不支持该指令集；若自行编译时使用的是Rosetta转译的x86环境，编译产物依旧是x86版本，会触发相同报错。

可行操作步骤 #

• 第一步：完全卸载现有jax相关包
  执行命令：pip uninstall -y jax jaxlib
  如果你使用conda环境，先确认你的conda是arm64原生版本，若之前安装的是x86版conda，需要先卸载重装arm64版Miniconda/Anaconda。
• 第二步：选择任意一种方式安装适配版本
  • 方案1（conda用户优先选）：执行conda install -c conda-forge jax jaxlib，conda-forge源已预编译适配M1芯片的jax相关包，安装完成后即可正常导入。
  • 方案2（pip用户选）：确认当前pip对应Python是arm64原生版本后，直接执行pip install jax jaxlib即可，官方已上架M1专属预编译包，无需自行编译。

自行编译注意事项 #

如果确实需要从源码编译，先执行arch命令确认终端输出为arm64，处于原生ARM运行环境，再按官方编译流程操作，避免编译出x86架构的不兼容包。

内容的提问来源于stack exchange，提问作者dcxst