如何通过编程终止Vertex AI Jupyter Notebook的执行？

1. 当前单元格终止后后续仍执行是预期行为吗？ #

是预期行为。不管是本地Jupyter Notebook还是Vertex AI托管的Notebook，单元格默认是独立执行单元——单个单元格的报错、终止只会结束当前单元格的运行，不会自动触发整个Notebook的执行停止。这是Jupyter生态的通用逻辑，并非Vertex AI特有的行为。

2. 终止整个Notebook执行的内置方法 #

你之前尝试的sys.exit()、assert False等方法仅作用于当前单元格，因为它们只是退出当前单元格的代码上下文，不会影响内核运行。要彻底终止整个Notebook的执行，可通过以下内置方式实现：

方法一：调用IPython内核API #

在需要终止的单元格中执行以下代码，直接关闭当前内核，后续单元格因无可用内核会停止执行：

from IPython import get_ipython
get_ipython().kernel.do_shutdown(restart=False)

若需要同时重启内核，可将restart参数改为True，仅终止执行则保持False即可。

方法二：界面操作终止 #

手动执行时，可直接点击Notebook界面顶部的方形「停止」图标，或在顶部菜单栏的「Kernel」选项中选择「Interrupt Kernel」（中断内核）、「Shutdown Kernel」（关闭内核），同样能立即终止所有后续执行。

替代繁琐变量判断的优化方案 #

如果不想在每个单元格开头写变量判断逻辑，可以自定义一个魔法函数，方便快速调用：

from IPython.core.magic import register_line_magic
from IPython import get_ipython

@register_line_magic
def stop_notebook(line):
    get_ipython().kernel.do_shutdown(restart=False)

定义完成后，在任意单元格输入%stop_notebook即可直接终止整个Notebook的执行，比重复编写判断逻辑简洁高效。

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