如何在完成脚本后断开 Google Colab

Question

我正在使用 Google Colab Pro+ 来训练计算机视觉模型。有时执行需要长达 8 小时，所以完成后我仍在分配内存和 GPU。我希望我有一行代码可以在训练结束后执行。有什么方法可以使用代码终止环境吗？

Answer 1

您应该能够通过运行以下命令断开运行时，

!kill $(ps aux | awk '{print }')

此命令将提取每个 运行 进程的 PID，然后杀死（或停止）它们。