Ubuntu 18.04安装TensorRT 7.2.1

mathlxj

于 2021-05-28 16:53:38 发布

阅读量2.3k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/mathlxj/article/details/117367574

ubuntu 同时被 2 个专栏收录

30 篇文章

订阅专栏

TensorRT

14 篇文章

订阅专栏

本文档详细介绍了如何下载、解压TensorRT 7.2.1，并在Ubuntu 18.04系统上配置环境变量，以及安装对应的Python版本和相关库（TensorRT、UFF、GraphSurgeon、ONNX GraphSurgeon）。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 下载TensorRT 7.2.1

下载地址

2. 解压缩

tar -zxvf TensorRT-7.2.1.6.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn8.0.tar.gz

3. 修改环境变量

vim ~/.bashrc
export LD_LIBRARY_PATH=/home/lixj/Downloads/trt-fasterrcnn/TensorRT-7.2.1.6/lib:$LD_LIBRARY_PATH

4. 安装对应python版本

cd TensorRT-7.2.1.6 python
pip install tensorrt-7.2.1.6-cp36-none-linux_x86_64.whl
cd ../uff/
pip install uff-0.6.9-py2.py3-none-any.whl
cd ../graphsurgeon/
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl
cd ../onnx_graphsurgeon/
pip install onnx_graphsurgeon-0.2.6-py2.py3-none-any.whl