原材料准备
tensorrt:TensorRT-7.2.3.4.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn8.1
onnx:onnx-1.8.1
onnx2tensorrt:onnx-tensorrt-release-7.2.1
# onnx 安装
https://github.com/onnx/onnx/releases
mkdir build
cd build
cmake ..
make
make install
# onnx2tensorrt 安装
https://github.com/onnx/onnx-tensorrt/releases
cd onnx-tensorrt
mkdir build
cd build
# /path/to/TensorRT-8.0.0.3改成自己的TensorRT绝对路径
cmake .. -DTENSORRT_ROOT=/path/to/TensorRT-8.0.0.3
make
make install
安装 成功!
查看onnx-tensorrt版本,在此处有与TensorRT之间的版本关系,里面thirdparty下有onnx的版本,需保持对应。