TensorRT 4.0没有这个,只能自己产生。5.0就有了。
git clone --recursive https://github.com/onnx/onnx-tensorrt.git rm -rf build mkdir build cd build cmake .. -DTENSORRT_ROOT=/home/quantum6/TensorRT-4.0.1.6 make -j8 sudo make install
注意,TensorRT使用绝对路径。
如果遇到错误,参考:
《onnx-tensorrt:builtin_op_importers.cpp:628:5: error: ‘IIdentityLayer’ is not a member of ‘nvinfer1’》
https://blog.csdn.net/quantum7/article/details/88781720