栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

BEVFusion-复现问题:CUDA error: no kernel image is available for execution on the device

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

BEVFusion-复现问题:CUDA error: no kernel image is available for execution on the device

问题:

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

解决思路,
本以为是显存问题,debug后发现似乎不是
按照提示设置

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

也不对,后来在网上查询感觉应该是CUDA算力在编译的时候没有匹配.

最终解决方案:
由于本人使用1080Ti,对应算力61
在setup25行加入

"-gencode=arch=compute_61,code=sm_61",

重新编译,就没有问题了.

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/983788.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号