longcw / yolo2-pytorch

YOLOv2 in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

pangyanhui923 opened this issue · comments

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

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.

me too

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 "