build tensorrt 自定义算子时出现cub/cub.cuh:No such file or directory #include "cub/cub.cuh"

[L-Carson]

安装 Model Converter

编译 TensorRT 自定义算子

mkdir -p build && cd build
cmake .. -DMMDEPLOY_TARGET_BACKENDS="trt"
make -j$(nproc) && make install
出现cub/cub.cuh:No such file or directory #include "cub/cub.cuh"问题

[zerollzeng]

Looks like this is not related to TRT?

I can still give some suggestions here: If you the cub is not come with your cuda package, you can build it on your own, see https://github.com/NVIDIA/cub

[L-Carson]

Looks like this is not related to TRT?
The problem I encountered while compiling the Tensorrt custom operator, so I described it with TRT
I can still give some suggestions here: If you the cub is not come with your cuda package, you can build it on your own, see https://github.com/NVIDIA/cub

thank you，I had tried to build it by myself, but I finded it support CUDA Toolkit >=11.0, while my system is jetson nano which CUDA Toolkit is 10.2, so I build it failed.

[zerollzeng]

I think you should upgrade the Jetpack.

[ttyio]

closing since no activity for more than 3 weeks, pls reopen if you still have question, thanks!