通过以上步骤,你应该能够定位并解决“nms_impl: implementation for device cuda:0 not found”的问题。如果问题仍然存在,可能需要更详细地检查你的代码和配置,或寻求来自开发社区的帮助。
RuntimeError: nms_impl: implementation for device cuda:0 not found.关注 AI韬哥 RuntimeError: nms_impl: implementation for device cuda:0 not found. 原创 AI韬哥 2023-05-18 17:04:54 ©著作权 文章标签 python 重新安装 文章分类 JavaScript 前端开发 卸载mmcv-full后重新安装。赞 收藏 评论 分...
Never mind found it at:https://github.com/open-mmlab/mmcv/blob/d9e10e11846d911e8354cd024967d3a17a88083c/mmcv/ops/nms.py#L127 inds=NMSop.apply(boxes.to('cpu'),scores.to('cpu'),iou_threshold,offset,score_threshold,max_num)
I found solution. I installed mmcv with onlypip install mmcv. However I think it is wrong way to install. Solution: When I uninstall and install withpip install mmcv-full==1.5.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html, then problem is solved. It works...