在Databricks Notebook中重启Python解释器仍报错的问题

分步排查方案 #

• 拆分代码单元格执行
  把安装包和导入代码拆成三个独立单元格分步执行：

  1. 第一个单元格仅执行安装命令：

     %pip install optree
     

  2. 第二个单元格单独执行重启命令：

     dbutils.library.restartPython()
     

     等待Notebook状态栏显示“已连接”（内核重启完成）后，再操作第三个单元格。
  3. 第三个单元格执行导入代码：

     import numpy as np
     from sklearn.preprocessing import MinMaxScaler
     from tensorflow.keras.models import Sequential
     from tensorflow.keras.layers import LSTM, Dense
     

• 排查集群库依赖冲突

  1. 进入Databricks集群管理页面，查看「库」标签页的预安装库列表，检查是否存在与optree版本冲突的库。
  2. 若发现冲突库，可先卸载冲突库后重新安装optree；或创建全新的测试集群，排除原有集群环境的干扰。

• 指定optree版本安装
  部分版本兼容性问题可能导致安装不完整，尝试指定稳定版本安装：

  %pip install optree==0.10.0
  

  安装完成后重复上述重启和导入步骤。

• 验证安装状态
  内核重启后，先执行以下命令确认optree是否安装成功：

  import pkg_resources
  print(pkg_resources.get_distribution("optree").version)
  

  若能正常输出版本号，说明安装有效，再执行后续导入操作即可。

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