Tensorflow 报错 Could not load dynamic library ‘libnvinfer.so.7‘ 解决方法

前言

运行 Tensorflow 之后发现一直有个报错,然后上网一查,原来是需要装一个叫 TensorRT 的库,这个库是 Nvidia 的。

PS:前提要先安装 CUDA 和 cuDNN

需要安装的同学上网一 Google 就出来了,不过需要注册 Nvidia 的一个账户。

我装的是最新版的 8.5 的 Tar 版本,包很大1G多,这个网页还附带了安装教程。

8.5问题就出在安装完成之后还是报错,然后仔细排查了一下问题,发现 Tensorflow 的报错是这样的。

errorTensorflow 找不到这个 libnvinfer.so.7 的文件,然后我们安装 lib 里面是这样的。

lib
也就是 Tensorflow 2.10 默认的 libnvinfer.so 的版本是 7,而不是我们装的 8,但是点开 7 的下载链接跳出的是一个调查问卷,然后开始自己瞎改。

surver

解决方法

直接复制两个 libnvinfer.solibnvinfer.so.7 就可以。

solution

cp libnvinfer_plugin.so libnvinfer_plugin.so.7
cp libnvinfer.so libnvinfer.so.7

然后训练模型就不会再爆这个错误了。

finish

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年9月7日
下一篇 2023年9月7日

相关推荐