Ubuntu18.04安装tensorrt,解决libcudnn.so.7: cannot open shared object file: No such file or directory

0、提前安装了这个(不安装好像也不是很影响,可以忽略)

$ pip3 install 'pycuda>=2017.1.1'

1、查看cuda和cudnn的版本号:
cuda:

$ cat /usr/local/cuda/version.txt

在这里插入图片描述
cudnn:

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

在这里插入图片描述
2、去nvida官网下载对应的tensorrt版本:
进入下载链接:https://developer.nvidia.com/nvidia-tensorrt-7x-download
点击Download Now(需要登录英伟达账号,没有的注册一个)
选择下载的版本
完成问卷调查
选择同意协议
根据自己的系统版本和CUDA版本,选择安装包,如图所示(如果是完整安装,建议选择Tar File Install Packages,我选择的是TensorRT7.0.0.11)
在这里插入图片描述
3、下载完tar包之后,选择一个地址进行压缩(我在本地建立了一个Tensorrt文件夹,也可以不建),之后会有一个带版本号的文件夹,看是否与自己的电脑匹配:

在这里插入图片描述
4、安装python tensorrt heel文件
进入解压后的文件夹找到python文件夹:
在这里插入图片描述
在这里插入图片描述
在这打开终端(或者自己 cd 输入命令进来该文件夹)输入以下命令,选择自己需要的.whl文件进行安装,修改相应的文件名字即可。(python 3 以下的版本使用pip2)

$ sudo pip3 install tensorrt-7.0.0.11-cp36-none-linux_x86_64.whl

5、安装python UFF whell文件(只有使用Tensorflow时需要)
进入解压后的文件夹找到uff文件夹,在此打开终端输入命令:

$ sudo pip3 install uff-0.6.5-py2.py3-none-any.whl

6、安装python graphsurgeon wheel

$ sudo pip3 graphsurgeon-0.4.1-py2.py3-none-any.whl

7、添加环境变量(如果出现libcudnn.so.7: cannot open shared object file: No such file or directory,同样是执行这一步)
编辑.bashrc文件,将TensorRT lib库绝对路径添加到环境变量LD_LIBRARY_PATH:

$ gedit ~/.bashrc

将:export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/Tensorrt/TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn7.6/TensorRT-7.0.0.11/lib复制进去

注意
1)/home/Tensorrt/TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn7.6/TensorRT-7.0.0.11是你自己的TensorRT lib库的路径;
2)在添加环境变量LD_LIBRARY_PATH=$LD_LIBRARY_PATH 的时候“=”前后不要有空格,不然环境变量添加不成功,报libcudnn.so.7: cannot open shared object file: No such file or directory这个错的小伙伴可以检查下是不是这里写错了,我就是当时自己输的有空格,明明添加了环境变量还一直报错没找到原因。

激活.bashrc文件:

$ source ~/.bashrc

8、检查是否安装成功,打开终端输入:

$ python3

然后执行以下命令,没报错就是安装成功了:

>>> import tensorrt
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值