配置YOLOX的时候,C 报错:/home/tkc/torch2trt/trt8/torch2trt/plugins/interpolate.cpp:6:10: fatal error: NvInfer.h: 没有文件或目录;subprocess.CalledProcessError: Command ‘[‘ninja’, ‘-v’]’ returned non-zero exit status 1.
NvInfer.h: 没有文件或目录
似乎缺少头文件。找了一波issue之后发现setup.py有bug, setup中的trt_inc_dir和trt_lib_dir固定返回路径,配置过程路径不一定一致。 该方法负责指向tensorRT,但这是一条固定的路径。需要改变自己的路径。必须从/home开始指向,否则报错!已经安装了torch2trt也要卸载干净。
再次安装带插件torch2trt安装完成后。