Ubuntu16.04下Cuda10.2的Tensorrt7.0安装及填坑指南

上一篇博客重新安装显卡驱动440和cuda10.2版本安装好了cuda10.2，那么这一篇讲述如何安装tensorrt7.0及遇到的坑。

安装tensorrt7

先去官网下载ubuntu16.04，cuda10.2对应的tensorrt7版本，==建议先把deb版本和tar版本都下载下来存好，==如下图

这里注意的是如果cuda是用deb版本安装的，那么tensorrt就要使用deb版本安装，否则会有可能报错。

下载下来后根据官网提供的安装指导安装指导安装即可。

第一步，运行以下command：

os="ubuntu1x04"
tag="cudax.x-trt7.x.x.x-ga-yyyymmdd"
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub

sudo apt-get update
sudo apt-get install tensorrt cuda-nvrtc-10-2

对应的截图如下：


第二步：
如果是Python2.7环境，运行：

sudo apt-get install python-libnvinfer-dev

将会安装上python-libnvinfer
如果是Python3.x环境，运行：

sudo apt-get install python3-libnvinfer-dev

将会安装上python3-libnvinfer
如果想使用TensorFlow的tensorrt，运行：

sudo apt-get install uff-converter-tf

将会安装上graphsurgeon-tf

第三步，验证是否安装成功，运行：

dpkg -l | grep TensorRT

出现如下图，表示安装成功。

填坑1

然而，import tensorrt却报错了，提示ModuleNotFoundError: No module named ‘tensorrt’，如下图：

这什么鬼。。。。
解决办法：
解压我们刚才下载的tar安装包，cd到xxxx/Tensorrt/Pythpon/路径下，下面有几个whl文件，pip安装对应的whl文件即可，
再次import tensorrt就不会报错了。

填坑2

import uff 报错

解决办法：
cd到xxxx/Tensorrt/uff/下，pip安装即可。

以上一般就可以正常使用了，在/usr/local/下会生成一个tensorrt文件夹，里面包含bin，data，samples四个文件夹，samples文件夹中是官方例程的源码。

https://blog.csdn.net/zong596568821xp/article/details/86077553