yolo2-pytorch icon indicating copy to clipboard operation
yolo2-pytorch copied to clipboard

cudaCheckError() failed : no kernel image is available for execution on the device

Open pangyanhui923 opened this issue 6 years ago • 3 comments

pangyanhui923 avatar Jun 02 '18 13:06 pangyanhui923

Hi, i use GPU for training,the following error occurred: cudaCheckError() failed : no kernel image is available for execution on the device
but it works when using cpu.

hh99566 avatar Jun 06 '18 14:06 hh99566

me too

michaelyyq avatar Mar 18 '19 13:03 michaelyyq

I solved the error. CUDA_ARCH="-gencode arch=compute_30,code=sm_30
-gencode arch=compute_35,code=sm_35
-gencode arch=compute_50,code=sm_50
-gencode arch=compute_52,code=sm_52
-gencode arch=compute_60,code=sm_60
-gencode arch=compute_61,code=sm_61
-gencode arch=compute_70,code=sm_70 "

michaelyyq avatar Mar 18 '19 13:03 michaelyyq