问题:在使用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
-f https
://download
.openmmlab
.com
/mmcv
/dist
/index
.html