会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
王冰冰
博客园
首页
新随笔
联系
订阅
管理
2024年5月15日
pytorch调试时CUDA报错,如何精确定位
摘要: 由于pytorch的python语句执行和CUDA后端实际上是异步执行的,所以当CUDA runtime报错时,并不一定能正确地反映在python抛出的错误语句中,因此,可以添加一个环境变量来实现: CUDA_LAUNCH_BLOCKING=1 这里再补充一些关于cuda和pytorch异步执行的知
阅读全文
posted @ 2024-05-15 20:15 王冰冰
阅读(52)
评论(3)
推荐(0)
编辑
公告