ubuntu20.04 安装onnx-tensorrt

1、克隆工程

git clone -b 7.1 https://github.com/onnx/onnx-tensorrt.git
cd onnx-tensorrt

克隆后onnx-tensorrt/third_party/onnx中内容是空的,需要下载与tensorrt7.1匹配的onnx1.6。将下载后的内容,解压放置到onnx-tensorrt/third_party/

wget https://github.com/onnx/onnx/releases/download/v1.6.0/onnx-1.6.0.tar.gz
tar -xzvf onnx-1.6.0.tar.gz
mv onnx-1.6.0 onnx

2、编译安装

cd onnx-tensorrt
mkdir build
cd build

cmake -DCMAKE_BUILD_TYPE=RELEASE -DCMAKE_INSTALL_PREFIX=~/pgf/deploy/onnx-tensorrt/build/installed -DTENSORRT_ROOT=/home/oem/software/TensorRT-7.2.3.4 -DBUILD_ONNX_PYTHON=OFF ..



 make -j4
 make install -j4

3、添加环境到工程中或环境变量中

gedit ~/.bashrc

该onnx_tensorrt版本可以添加到相应工程编译文件中
export PATH=/home/oem/pgf/deploy/onnx-tensorrt/build/installed/bin:$PATH
export LD_LIBARARY_PATH=/home/oem/pgf/deploy/onnx-tensorrt/build/installed/lib:$LD_LIBARARY_PATH

source ~/.bashrc

onnx2trt -V

你可能感兴趣的:(板端物联网,開發環境配置,git)