gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-04 00:04:36 25 4
gpt4 key购买 nike

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

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

最佳答案

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

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

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

关于python - 如何在出现以下 CUDA 错误 'Cuda assert fails: device-side assert triggered' 后重置 Colab?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54585685/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com