配置MMdetection踩过的坑:RuntimeError: nms is not compiled with GPU support

这个坑查阅了很多博客一般都说是pytorch版本和mmcv版本不兼容导致,因此我在配置过程中更换了很多版本的pytorch版本,仍然没有用,最终发现是mmcv下build文件夹没有删除,尤其是从同事或同学拷贝过来的代码,一定要先删除该文件夹,然后再进行环境配置。

版权声明:本文为CSDN博主「CV蒸汽机」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/dabian1793/article/details/114261667

CV蒸汽机

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

Anaconda虚拟环境中安装CUDA

解决问题:当我们运行不同的代码时,可能会用到不同版本的Pytorch/Tensorflow,不同版本的Pytorch/Tensorflow在使用GPU时,对应不同的CUDA版本&#xff0

目标检测:学习yolo版本一

笔记
论文的脉络
先对比了和一些 其他工作的比较,比如 比 fast rcnn 更加能利用 全局的信息。
牵扯到的基础知识
iou
mAP
2. 统一的检测流程
提出了一个想法:每一个 box 如果 有