欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu18.04安装tensorrt

程序员文章站 2022-06-05 20:42:00
...

解压下载的文件

添加环境变量

vim ~/.bashrc
export LD_LIBRARY_PATH=/home/liguiyuan/software/TensorRT-7.0.0.11/lib:$LD_LIBRARY_PATH
source ~/.bashrc

进入解压目录 使用pip安装

TensorRT
UFF
graphsurgeon

验证安装

>>> import tensorrt
>>> tensorrt.__version__
'7.0.0.11'

错误

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

解决办法

sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/lib*  /usr/lib/
sudo ldconfig #立即生效