RuntimeError:cuda runtime error unrecognized error code【解决方法】

今天爱分享给大家带来RuntimeError:cuda runtime error 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