`Deformable-DETR`报错error in ms_deformable_im2col_cuda: no kernel image is available for execution on the device
问题描述
在使用Deformable-DETR
时需要先编译CUDA操作符,如下,但是在编译后,运行训练任务时,程序报error in ms_deformable_im2col_cuda: no kernel image is available for execution on the device
,多方搜索后找到答案:make.sh · Issue #10 · fundamentalvision/Deformable-DETR (github.com)
解决方案
-
记得先删掉之前编译的
-
先
cd /usr/local/
,查看cuda的路径:
- cd回
Deformable-DETR
,编译前加一句export CUDA_HOME=/usr/local/cuda-11.1
(根据自己的路径改),即可!