1 TensotRT7 安装 ONNX_tensorrt教程
1 git clone --recursive -b 7.0 https://github.com/onnx/onnx-tensorrt.git
2 cd onnx-tensorrt
3 mkdir build
4 cd build
5 cmake .. -DTENSORRT_ROOT=<tensorrt_install_dir> -DGPU_ARCHS="61"
这一步可能报错:
Could NOT find Protobuf (missing: Protobuf_INCLUDE_DIR)
解决办法:sudo apt-get install libprotobuf-dev protobuf-compiler
6 make -j 12
make -j 12
这一步如果报错
numeric_limits’ is not a member of ‘std’
表示出错的onnx_utils.hpp头文件没有包含
#include <stdexcept> #include <limits>
7 sudo make install
2 TensotRT4 安装 ONNX_tensorrt5.1教程
1 git clone --recursive -b v5.0 https://github.com/onnx/onnx-tensorrt.git
2 cd onnx-tensorrt
3 mkdir build
4 cd build
5 cmake .. -DTENSORRT_ROOT=<tensorrt_install_dir> -DGPU_ARCHS="61"
可能报错:
onnx-tensorrt:builtin_op_importers.cpp:628:5: error: ‘IIdentityLayer’ is not a member of ‘nvinfer1’
解决方法:
如果在TensorRT 4.0上编译,就有这个错误;5.0就没有(集成了)。
因此注释报错代码所在的整个函数,重新编译即可。
6 make -j8
7 sudo make install