天天看點

編譯onnx-tensorrt産生libnvonnxparser.so

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

繼續閱讀