我在 Google Colab 上使用 Pytorch 运行我的 Jupyter Notebook。在我收到“Cuda 断言失败：设备端断言触发”之后，我无法运行任何其他使用我的 pytorch 模块的代码。有谁知道如何重置我的代码，以便我之前工作的 Pytorch 函数仍然可以运行？我已经尝试实现，CUDA_LAUNCH_BLOCKING=1但我的代码仍然无法工作，因为断言仍然被触发！
1 回答
一只甜甜圈
TA贡献1836条经验 获得超5个赞
您需要重置 Colab 笔记本。要运行以前可以工作的现有 Pytorch 模块，您必须执行以下操作：
转到工具栏中的“运行时”
单击“重新启动并运行所有”
这将重置您的 CUDA 断言并刷新模块，以便您可以再次尝试避免错误！
添加回答
举报
0/150
提交
取消