当前位置: 代码迷 >> 综合 >> onnx-tensorrt安装
  详细解决方案

onnx-tensorrt安装

热度:63   发布时间:2023-10-16 17:02:15.0

github地址
安装protobuf和tensorRT8.0.1.6
接着执行以下指令

git clone https://hub.fastgit.org/onnx/onnx-tensorrt.git
cd onnx-tensorrt
mkdir build && cd build
#<path_to_trt>是tensorRT的安装地址
cmake .. -DTENSORRT_ROOT=<path_to_trt> && make -j8
#后面为tensorrt的lib库位置
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/zty/TensorRT-8.0.1.6/lib