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编译过。

装了好久,哭哭。

posted @ 2021-10-19 11:22  John_Ran  阅读(1525)  评论(0编辑  收藏  举报