Jupyter Notebook执行cell后括号无编号为空的原因及解决办法

• 核心原因是Jupyter Notebook/Lab的内核进程出现了未捕获的崩溃或挂起，TensorFlow 2.5.0和Python 3.7的环境组合下，最常见的触发场景是显存OOM（内存不足）、底层CUDA驱动版本不匹配导致的静默崩溃，或者训练时数据加载线程死锁。
• 单元格正常运行时左侧会显示[*]代表正在执行，运行结束后会替换为对应执行编号；如果内核直接异常退出，但没有向前端返回运行结束或报错的信号，前端就会停留在空白状态，既没有星号也没有编号。

1. 排查环境兼容性问题 #

• 首先确认安装的TensorFlow 2.5.0对应的CUDA和cuDNN版本是否匹配：TensorFlow 2.5.0官方要求CUDA版本为11.2、cuDNN版本为8.1，版本不匹配会大概率出现训练时内核静默崩溃的问题。
• 执行pip list | grep tensorflow确认安装的是CPU版还是GPU版，如果是GPU版可以先切换为CPU版运行训练，排查是否为GPU相关依赖导致的问题。

2. 排查显存溢出问题 #

• 训练脚本开头添加显存动态分配代码，避免TensorFlow默认占用全部显存触发OOM：

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

• 适当调小batch size，减少单步训练的显存占用。

3. 恢复单元格正常显示 #

• 如果已经出现空白状态，先在界面菜单栏点击「Kernel」-「Interrupt」尝试中断内核，无效的话选择「Restart Kernel」重启内核即可恢复单元格状态显示。
• 可以将Jupyter Notebook升级到6.4及以上版本，该版本优化了内核状态检测逻辑，能更准确捕获内核崩溃信号，减少空白状态出现的概率。

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