前言
运行 Tensorflow 之后发现一直有个报错,然后上网一查,原来是需要装一个叫 TensorRT 的库,这个库是 Nvidia 的。
PS:前提要先安装 CUDA 和 cuDNN
需要安装的同学上网一 Google 就出来了,不过需要注册 Nvidia 的一个账户。
我装的是最新版的 8.5 的 Tar 版本,包很大1G多,这个网页还附带了安装教程。
libnvinfer.so.7
的文件,然后我们安装 lib
里面是这样的。
也就是 Tensorflow 2.10 默认的 libnvinfer.so
的版本是 7,而不是我们装的 8,但是点开 7 的下载链接跳出的是一个调查问卷,然后开始自己瞎改。
解决方法
直接复制两个 libnvinfer.so
到 libnvinfer.so.7
就可以。
cp libnvinfer_plugin.so libnvinfer_plugin.so.7
cp libnvinfer.so libnvinfer.so.7
然后训练模型就不会再爆这个错误了。
文章出处登录后可见!
已经登录?立即刷新