Azi*_*bro 3 python pytorch tensor google-colaboratory
我在 Google Colab 上使用 Pytorch 运行我的 Jupyter Notebook。在我收到“Cuda 断言失败:设备端断言触发”之后,我无法运行任何其他使用我的 pytorch 模块的代码。有谁知道如何重置我的代码,以便我之前工作的 Pytorch 函数仍然可以运行?
我已经尝试实现,CUDA_LAUNCH_BLOCKING=1但我的代码仍然无法工作,因为断言仍然被触发!
您需要重置 Colab 笔记本。要运行以前可以工作的现有 Pytorch 模块,您必须执行以下操作:
这将重置您的 CUDA 断言并刷新模块,以便您可以再次尝试避免错误!
| 归档时间: |
|
| 查看次数: |
9329 次 |
| 最近记录: |