Fatal error: cub/cub.cuh: No such file or directory #include "cub/cub.cuh"

I solved by including path /opt/nvidia/deepstream/deepstream-6.0/sources/TensorRT/third_party/cub
to include_directories in CmakeLists.txt