问题:在使用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

在使用mmdetection进行训练时遇到`RuntimeError: CUDA error: invalid device function`的问题。官方issue指出可能是mmcv版本不兼容。解决方案是更新到与环境匹配的mmcv版本。环境配置包括Ubuntu 16.04、Anaconda3、Python 3.7.7、CUDA 10.0、cuDNN 7.6.4.3、Pytorch 1.6.0。安装mmdetection和mmcv可参照官方教程,特别是mmcv应使用特定命令安装。
最低0.47元/天 解锁文章
3053

被折叠的 条评论
为什么被折叠?



