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编译过。
装了好久,哭哭。