无法导入';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种可能性
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驱动程序: