在使用mmdet框架并且GPU为A100时,会报错RuntimeError:CUDA error:no kernel image is available for execution on the device。但是可以正常使用pytorch,即pytorch版本的算力是匹配上显卡的算力的。
解决措施是:使用压缩包安装mmdet框架,在执行setup之前,需要在setup第25行添加本机显卡对应的算力设置,否则CUDA算力在编译的时候没有匹配
Ge TITAN X 对应算力为52 | RTX 3090 3080 对应算力为86 | A100 对应算力为80
添加"-gencode=arch=compute_52,code=sm_52",
添加"-gencode=arch=compute_86,code=sm_86",
python setup.py develop // 最后安装mmdet,编译通过