如何在出现以下 CUDA 错误“Cuda 断言失败：设备端断言触发”后重置 Colab？

Question

我在 Google Colab 上使用 Pytorch 运行我的 Jupyter Notebook。在我收到“Cuda 断言失败：设备端断言触发”之后，我无法运行任何其他使用我的 pytorch 模块的代码。有谁知道如何重置我的代码，以便我之前工作的 Pytorch 函数仍然可以运行？

我已经尝试实现，CUDA_LAUNCH_BLOCKING=1但我的代码仍然无法工作，因为断言仍然被触发！

Answer 1

您需要重置 Colab 笔记本。要运行以前可以工作的现有 Pytorch 模块，您必须执行以下操作：转到工具栏中的“运行时”单击“重新启动并运行所有”这将重置您的 CUDA 断言并刷新模块，以便您可以再次尝试避免错误！