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

Question

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

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

Answer 1

您需要重置Colab笔记本。要运行以前曾经使用过的现有Pytorch模块，您必须执行以下操作：

• 转到工具栏中的“运行时”
• 点击“重新启动并全部运行”

这将重置您的CUDA断言并刷新模块，以便您可以避免错误再次发生！