Conda重装jaxlib后报无version属性AttributeError如何解决

尝试升级jaxlib到最新版本时操作失败，执行以下步骤重装：

• 运行conda remove --force jaxlib强制卸载原有jaxlib
• 运行conda install -c conda-forge jaxlib从conda-forge频道安装jaxlib
  安装完成后运行程序抛出如下报错，暂未定位问题成因，也未找到有效解决方法：

AttributeError: module 'jaxlib' has no attribute 'version'

这个错误是强制卸载导致的包残留+jax/jaxlib版本强绑定不匹配引发的：

• 执行conda remove --force jaxlib时，强制移除操作跳过了依赖校验，本地原本和旧版jaxlib匹配安装的jax包没有被同步卸载。而conda-forge频道拉取的新版jaxlib已经调整了版本属性的导出逻辑，本地留存的旧版jax启动时会主动读取jaxlib.version字段做兼容性校验，找不到对应属性就会直接抛出这个AttributeError。
• 小概率情况是强制卸载时旧版jaxlib的本地文件没有被完全清理，Python导入时加载到了残缺的旧版文件，也会触发同类报错。

按顺序操作即可解决：

• 先彻底清理所有jax相关包的残留，不要加force参数，避免卸载不完整：

  conda remove jax jaxlib
  

  清理完成后分别执行conda list | grep jax、pip list | grep jax检查，确认输出里没有jax、jaxlib相关条目，没有残留才算清理完成。
• 不要单独安装jaxlib，必须同时安装版本匹配的jax和jaxlib，从根源避免版本错配：

  conda install -c conda-forge jax jaxlib
  

• 安装完成后做个简单验证，进入Python交互环境执行以下代码，能正常输出版本号就说明修复完成：

  import jaxlib
  print(jaxlib.__version__)
  

补充提示：jax和jaxlib是强依赖绑定的关系，永远不要单独升级、单独安装其中某一个，否则很容易出现各类导入错误、属性不存在的兼容问题。

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