2

我已安装Tensorflow r1.14并想使用TF-TRT. 但是，会出现以下错误：

“ModuleNotFoundError：没有名为‘tensorflow.contrib.tensorrt’的模块”

运行示例代码时。发生同样的错误Tensorflow r1.13。所以我的问题是我需要tensorflow.contrib.tensorrt单独安装库吗？如果是，如何？

TensorRT另外，我可以成功运行例如的示例代码sampleINT8。单击此处查看我成功运行的示例代码。

这使我相信TensorRT安装正确。但是，TF-TRT仍然不起作用。

任何帮助将不胜感激！

4

4 回答 4

3

在 TF 1.14 中，TF-TRT 从 contrib 移至核心。

您需要像这样导入它： from tensorflow.python.compiler.tensorrt import > trt_convert as trt

https://github.com/tensorflow/tensorrt/blob/master/tftrt/examples/image-classification/image_classification.py#L22

这是 Linux 的正确答案。

但是，如果您使用的是 Windows：目前 Windows 不支持 TensorRT Python API（以及因此 TF-TRT），因此 TensorFlow python 包不是使用 TensorRT 构建的。

于 2019-11-14T19:03:29.723 回答
2

在 TF 1.14 中，TF-TRT 从 contrib 移至核心。

您需要像这样导入它： from tensorflow.python.compiler.tensorrt import trt_convert as trt

https://github.com/tensorflow/tensorrt/blob/master/tftrt/examples/image-classification/image_classification.py#L22

于 2019-06-11T19:00:22.427 回答
0

2 种可能性

  1. 您是否安装了 tensorflow-gpu 而不是 tensorflow？
  2. 从您的屏幕截图看来，您使用的是 Windows。我有同样的问题。TF windows 发行版中似乎没有tensorrt模块，contrib但是 linux 有它（我试过 1.13.1）。
于 2019-07-16T21:52:39.307 回答
0

为了能够，import tensorflow.contrib.tensorrt您需要tensorflow-gpu 在您的系统上安装 >= 1.7 版本。也许您可以尝试tensorflow-gpu使用以下命令安装库：

pip install tensorflow-gpu

也可以查看GPU 文档的 Windows 部分。另外，我会尝试使用以下命令更新您的tensorflow版本：

pip install --upgrade tensorflow

以确保您在那里也是最新的。查看TensorFlow 文档的这一部分以获得更多支持。

希望这会有所帮助！

于 2019-04-04T19:52:30.307 回答