RuntimeError: cuda runtime error (98) : unrecognized error code

导读

在使用mmdetection训练目标检测模型的时候，报

RuntimeError: cuda runtime error (98) : unrecognized error code at mmdet/ops/sigmoid_focal_loss/src/sigmoid_focal_loss_cuda.cu:139

环境信息

• Ubuntu16.04
• pytorch：1.3.1
• torhvision：0.4.2
• mmdet：1.0rc1+173ef21
• mmcv：0.2.14
• cuda：10.0
• cudnn：7.6

解决办法

安装与你pytorch相匹配的CUDA Toolkit，我上面的环境可以使用如下命令进行安装

conda install pytorch=1.3 torchvision cudatoolkit=10.0 -c pytorch

注意：安装完成之后需要重新编译项目和mmdet