CUDA kernel errors might be asynchronously reported at some other API call 错误解决

这篇具有很好参考价值的文章主要介绍了CUDA kernel errors might be asynchronously reported at some other API call 错误解决。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

Pytorch项目报错:

CUDA kernel errors might be asynchronously reported at some other API call

在运行基于pytorch的深度学习项目时,有时候会遇到上述错误,并且在报错时没有定位到正确的位置。


原因分析:

这里查阅了很多网上的相关资料,说是分类数目和模型里的实际分类数目不匹配,大家可以仔细查看一下这个。也有说是在计算logits之前先进性softmax归一化,大家也可以试试这个方法。

我仔细检查自己的项目分类数目,发现并不是上述原因,因此需要先定位到实际导致错误的代码位置。


解决方案:

我们的目的是定位到实际导致报错的代码位置,只需要在最开头添加:

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'

这时候报错会定位到实际位置,大家对症下药解决即可。文章来源地址https://www.toymoban.com/news/detail-543328.html

到了这里,关于CUDA kernel errors might be asynchronously reported at some other API call 错误解决的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包