mmyolo

[LittleNoob2333]

python 3.8
pytorch 1.10.0+cu111
cuda 11.1

在训练mmyolo下的小例子中，第10个epoch后出现
File "/home/wly/trtproject/mmyolo_tensorrt-main/mmcv/mmcv/ops/nms.py", line 27, in forward
inds = ext_module.nms(
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.