问题:在使用mmdetection做训练的时候,出现错误
RuntimeError: CUDA error: invalid device function
在官网的issue中看到是mmcv的版本不匹配问题,重新安装匹配的mmcv版本即可。
官网安装链接
环境版本
ubuntu16.04+Anaconda3+python3.7.7+cuda10.0+cuDNN7.6.4.3
pytorch1.6.0+python3.7
mmdetection的安装按照官网教程走即可,mmcv的安装用下面这句:
pip install mmcv-full==latest+torch1.6.0+cu101