RuntimeError: CUDA error: invalid device function【mmdetection错误】

it2025-04-01  21

问题:在使用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
最新回复(0)