目录
onnx转tensorrt方法
先把onnx转化为TensorRT的trt文件,然后让c++环境下的TensorRT直接加载trt文件,从而构建engine,本文主要讲解onnx转换至trt的过程。
tensorrt官网下载链接:
https://developer.nvidia.com/nvidia-tensorrt-download
我下载的版本:
TensorRT 8.2 GA for x86_64 Architecture
Zip Packages for Windows.
配置环境变量:
将下载好的文件解压到一个安静的磁盘位置,然后将“XX:\TensorRT-8.2.1.8\lib”添加环境变量。 环境变量位置: 此电脑图标–右键属性–高级系统设置–高级–环境变量–系统变量–path