配置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后,完成安装。