[Feature] implementation for device cuda:0 not found

open-mmlab / mmocr

OpenMMLab Text Detection, Recognition and Understanding Toolbox

https://mmocr.readthedocs.io/en/dev-1.x/

Apache License 2.0

4.29k stars 744 forks source link

[Feature] implementation for device cuda:0 not found #2018

Open wangzekun01 opened 8 months ago

wangzekun01 commented 8 months ago

What is the feature?

When I run the following command “python tools/train.py configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_1200e_icdar2015.py” I got “RuntimeError: modulated_deformable_im2col_impl: implementation for device cuda:0 not found.” Does anyone know how to deal with that?

Any other context?

No response