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