ubuntu18.04+cuda10.2+tensorrt8.4.1.5配置安装

xuan_gxi

已于 2022-08-14 10:43:39 修改

阅读量2.1k

点赞数 1

分类专栏：平时学习总结文章标签： linux python 深度学习

于 2022-07-25 22:22:39 首次发布

本文链接：https://blog.csdn.net/xuan_gap/article/details/125946440

版权

平时学习总结专栏收录该内容

4 篇文章 0 订阅

订阅专栏

下载tensorrt8.4.1.5

https://developer.nvidia.com/nvidia-tensorrt-8x-download
在这里插入图片描述 `

tar -zxvf TensorRT-8.4.1.5.Linux.x86_64-gnu.cuda-10.2.cudnn8.4.tar.gz
`
放在主目录下面

添加环境变量

sudo gedit ~/.bashrc
export LD_LIBRARY_PATH=/home/xuanxi/TensorRT-8.4.1.5/lib:$LD_LIBRARY_PATH
export LIBRARY_PATH=/home/xuanxi/TensorRT-8.4.1.5/lib::$LIBRARY_PATH
source ~/.bashrc

安装tensorrt的python包。进入tensorrt压缩之后的文件夹，然后进入里面的python文件夹

cd ./TensorRT-8.4.1.5/python

在这里插入图片描述 python文件夹里面有很多版本，使用pip安装自己对应的python版本
进入虚拟环境(!!!)

source activate torch1.7
pip install tensorrt-8.4.1.5-cp37-none-linux_x86_64.whl

(torch1.7) xuanxi@xuanxi-MS-7C02:~/TensorRT-8.4.1.5/python$ python
Python 3.7.9 (default, Aug 31 2020, 12:42:55) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorrt
>>> tensorrt.__version__

在这里插入图片描述这就安装成功了！！！

可以顺便安装其他两个包
1）安装 uff 包

cd ../uff    # 切换到 uff 文件夹
pip install uff-0.6.9-py2.py3-none-any.whl

测试是否安装成功：

which convert-to-uff  # 会输出安装路径

2）安装 graphsurgen 包

cd ../graphsurgeon     # 切换到 graphsurgeon 文件夹
pip install graphsurgeon-0.3.2-py2.py3-none-any.whl

安装 Polygraphy，或者通过 TensorRT/tools/Polygraphy 源码安装

python -m pip install colored polygraphy --extra-index-url https://pypi.ngc.nvidia.com

运行 ONNX 模型，保存输入输出

polygraphy run model_4.onnx --onnxrt --val-range [0,1] --save-inputs onnx_inputs.json --save-outputs onnx_outputs.json

运行 TRT 模型，载入 ONNX 输入输出，对比输出的相对误差与绝对误差

polygraphy run rvm_mobilenetv3_fp32_sim_modified.engine --model-type engine --trt --load-inputs onnx_inputs.json --load-outputs onnx_outputs.json --rtol 1e-3 --atol 1e-3

使用上面生成的 model.onnx 构建 TensorRT 引擎，使用 FP32 精度同时在 TensorRT 和 onnxruntime 中运行

输出所有层的计算结果作对比

polygraphy run model_4.onnx
–onnxrt --trt
–workspace 100000000
–save-engine=model_FP32_MarkAll.plan
–atol 1e-3 --rtol 1e-3
–verbose
–onnx-outputs mark all
–trt-outputs mark all
–trt-min-shapes ‘x:0:[1,1,28,28]’
–trt-opt-shapes ‘x:0:[4,1,28,28]’
–trt-max-shapes ‘x:0:[1,28,28]’
–input-shapes ‘x:0:[1,80,300]’
> result-run-FP32-MarkAll.txt

报错：libnvinfer.so.8: cannot open shared object file: No such file or directory
解压后进入TensorRT根目录：
（1）环境变量
sudo gedit ~/.bashrc
在文件末尾添加一行代码：

export LD_LIBRARY_PATH=/opt/TensorRT-8.4.2.4/lib:$LD_LIBRARY_PATH

关闭保存
source ~/.bashrc
（2）复制文件到系统路径(!!)
把TensorRT根目录中的/lib/下面的文件复制到 /usr/lib/下，
把TensorRT根目录中的/include/下面的文件复制到 /usr/include/下

xuanxi@xuanxi-MS-7C02:~/TensorRT-8.4.2.4$ sudo cp -r lib/* /usr/lib/
xuanxi@xuanxi-MS-7C02:~/TensorRT-8.4.2.4$ sudo cp -r include/* /usr/include/