tensorflow - libnvinfer.so.5：无法打开共享对象文件：没有这样的文件或目录

Question

我正在使用 Ubuntu 16.04 和 TensorFlow 1.13.1。现在我想集成 TensorRT 来改进我的模型的推理时间。我下载并解压了 TensorRT7 的 tar，并在其路径中安装了 uff 和 graphsurgeon 的 whls。我还将其路径添加到系统的 LD_LIBRARY_PATH。

但是，当我尝试导入 tensorflow.contrib.tensorrt 时，它给了我一个找不到文件的错误。我的 TensorRT7 的文件夹中没有 libnvinfer.so.5，而是一个 libnvinfer.so.7。

这是否意味着 TensorFlow 1.13.1 不支持 TensorRT7？我应该改用 TensorRT5 吗？

标签： tensorflowtensorrt