如何在出现以下 CUDA 错误 'Cuda assert fails: device-side assert triggered' 后重置 Colab?

How to reset Colab after the following CUDA error 'Cuda assert fails: device-side assert triggered'?

我正在 运行使用 Pytorch 在 Google Colab 上安装我的 Jupyter Notebook。在我收到 'Cuda assert fails: device-side assert triggered' 后,我无法 运行 任何其他使用我的 pytorch 模块的代码。有谁知道如何重置我的代码,以便我之前工作的 Pytorch 函数仍然可以 运行?

我已经尝试实施 CUDA_LAUNCH_BLOCKING=1 但我的代码仍然无法正常工作,因为 Assert 仍然被触发!

您需要重置 Colab 笔记本。对于 运行 以前可以工作的现有 Pytorch 模块,您必须执行以下操作:

  • 转到工具栏中的'Runtime'
  • 点击'Restart and Run all'

这将重置您的 CUDA 断言并清除模块,以便您可以再次尝试避免错误!