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