mmdetection安装问题(nms is not compiled with GPU support)


在按照官方的安装教程进行mmdetection安装的时候,出现了一些问题。我的环境信息如下:
最后报错信息如下:
从MMCV Compiler这儿来看,可以看见MMCV CUDA Compiler: not available。

原因是,我使用的mmcv-full的脚本是:

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.1/index.html

使用如下脚本安装就可以了:

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html

原因在官网上可以找到:

原因是mmcv-full只在pytorch 1.x.0编译过。

装了好久,哭哭。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM