无法导入';tensorflow.contrib.tensorrt';在tensorflow r1.14中

无法导入';tensorflow.contrib.tensorrt';在tensorflow r1.14中,tensorflow,tensorrt,Tensorflow,Tensorrt,我已经安装了Tensorflow r1.14,想使用TF-TRT。但是,出现以下错误: “ModuleNotFoundError:没有名为'tensorflow.contrib.tensorrt'的模块” 运行示例代码时。同样的错误发生在Tensorflow r1.13上。所以我的问题是我是否需要单独安装tensorflow.contrib.tensorrt库?如果是,如何进行 此外,我还可以成功运行TensorRT的示例代码,例如sampleINT8。单击此处查看成功运行的示例代码 这使我相信

我已经安装了
Tensorflow r1.14
,想使用
TF-TRT
。但是,出现以下错误:

“ModuleNotFoundError:没有名为'tensorflow.contrib.tensorrt'的模块”

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

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

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


任何帮助都将不胜感激

为了能够
导入tensorflow.contrib.tensorrt
,您需要在系统上安装
tensorflow gpu
版本>=1.7。也许您可以尝试安装带有以下组件的
tensorflow gpu
库:

pip安装tensorflow gpu

还可以查看的Windows部分。此外,我会尝试使用以下命令更新您的
tensorflow
版本:

pip安装--升级tensorflow

以确保您也能了解最新情况。查看的此部分以获得更多支持


希望这有帮助

在TF 1.14中,TF-TRT从contrib移到核心

您需要像这样导入它:
从tensorflow.python.compiler.tensorrt导入trt\u转换为trt

2种可能性

  • 您是否安装了tensorflow gpu而不是tensorflow
  • 从您的屏幕截图上看,您似乎正在使用Windows。我也有同样的问题。TF windows发行版中的
    contrib
    下似乎没有
    tensorrt
    模块,但linux有它(我尝试了1.13.1)

  • 在TF 1.14中,TF-TRT从contrib移至堆芯

    您需要这样导入它:从tensorflow.python.compiler.tensorrt import>trt\u转换为trt

    这是Linux的正确答案


    但是,如果您使用的是Windows:目前Windows不支持TensorRT Python API(因此TF-TRT),因此TensorFlow Python包不是用TensorRT构建的。

    TF.\uuuu版本为1.13.1,大于等于1.7。我还检查了文件夹“tensorflow\contrib”,没有名为tensorrt的子文件夹。我不认为pip安装会带来tensorrt。我现在还能做什么?也许您可以尝试直接从nvidia下载tensorrt驱动程序: