RuntimeError: nms is not compiled with GPU support

在用faster运行过程中报错:
RuntimeError: nms is not compiled with GPU support

1. 产生原因:

运行报错代码位置:

faster-rcnn/lib/model/rpn/proposal_layer.py 第147行

在这里插入图片描述

keep_idx_i = nms(proposals_single, scores_single.squeeze(1), nms_thresh)

2. 解决办法:

暂无解决办法

3. 怀疑可能原因:

1.配置环境的cuda版本和硬件显卡不匹配,环境配置如下图所示:
在这里插入图片描述
尝试过网上多种解决办法均无效果

4. 参考链接:

https://blog.csdn.net/weixin_44288510/article/details/115334372

猜你喜欢

转载自blog.csdn.net/weixin_44409833/article/details/127239084
NMS