CUDA kernel errors might be asynchronously reported at some other API call 错误解决
短信预约 -IT技能 免费直播动态提醒
Pytorch项目报错:
CUDA kernel errors might be asynchronously reported at some other API call
在运行基于pytorch的深度学习项目时,有时候会遇到上述错误,并且在报错时没有定位到正确的位置。
原因分析:
这里查阅了很多网上的相关资料,说是分类数目和模型里的实际分类数目不匹配,大家可以仔细查看一下这个。也有说是在计算logits之前先进性softmax归一化,大家也可以试试这个方法。
我仔细检查自己的项目分类数目,发现并不是上述原因,因此需要先定位到实际导致错误的代码位置。
解决方案:
我们的目的是定位到实际导致报错的代码位置,只需要在最开头添加:
import osos.environ['CUDA_LAUNCH_BLOCKING'] = '1'
这时候报错会定位到实际位置,大家对症下药解决即可。
来源地址:https://blog.csdn.net/qq_41018465/article/details/128655417
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341