BEVFormer_tensorrt
BEVFormer_tensorrt copied to clipboard
RuntimeError: modulated_deformable_im2col_impl: implementation for device cuda:0 not found.
I have the following problems when I run"sh samples/bevformer/plugin/base/pth2onnx.sh", how can I solve them? RuntimeError: modulated_deformable_im2col_impl: implementation for device cuda:0 not found.
I met this problem before, and I fix it with installing mmcv with FORCE_CUDA="1" and MMCV_WITH_OPS=1. Hope it helps you. https://github.com/DerryHub/BEVFormer_tensorrt/blob/303d3140c14016047c07f9db73312af364f0dd7c/docker/Dockerfile#L97
I installing mmcv with FORCE_CUDA="1" and MMCV_WITH_OPS=1,But meet this error also。 Can you give me onnx with custom plugin,I want to try run it on orin by trtexec(BSW if it can)~