yolo2-pytorch
yolo2-pytorch copied to clipboard
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 "