1

我在导入 Logger() 和 Builder() 时遇到错误

我在 Jetson AGX Xavier 上。

我也尝试过python shell。

import tensorflow.contrib.tensorrt as trt
TRT_LOGGER = trt.Logger(trt.Logger.INFO)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: module 'tensorflow.contrib.tensorrt' has no attribute 'Logger'

我也不能导入 trt.Builder() 。

Python 版本是 3.6.7

“dpkg -l | grep nvinfer”的输出给出了 Tensor RT 版本：

ii  libnvinfer-dev                             5.0.6-1+cuda10.0                                arm64        TensorRT development libraries and headers
ii  libnvinfer-samples                         5.0.6-1+cuda10.0                                all          TensorRT samples and documentation
ii  libnvinfer5                                5.0.6-1+cuda10.0                                arm64        TensorRT runtime libraries
ii  python-libnvinfer                          5.0.6-1+cuda10.0                                arm64        Python bindings for TensorRT
ii  python-libnvinfer-dev                      5.0.6-1+cuda10.0                                arm64        Python development package for TensorRT
ii  python3-libnvinfer                         5.0.6-1+cuda10.0                                arm64        Python 3 bindings for TensorRT
ii  python3-libnvinfer-dev 5.0.6-1+cuda10.0 arm64 Python 3 development package for TensorRT
4

2 回答 2

1

你导入的包import tensorflow.contrib.tensorrt as trt不是TensorRT，这是将TensorRT集成到TF中的包。该软件包没有您要查找的模块，例如 Logger 或 Builder。这个包有自己的 API，用于使用 TensorRT 优化 TF 模型。

更多将 TensorRT 集成到 TF 的链接：

如果您正在寻找直接在 TF 之外使用 TensorRT API，您应该像这样导入它：import tensorrt as trt.

于 2019-07-07T20:32:28.557 回答
0

我解决了这个问题。我所做的是：

我已经删除了 Tensorflow 1.13 并安装了 1.12。我删除了所有的虚拟环境。对 Jetpack 4.2 及其 Tensor RT 版本未做任何更改。

但我仍然找不到问题所在。

谢谢

于 2019-07-11T06:41:12.517 回答