Tensorflow> ModuleNotFoundError:没有名为'tensorflow.contrib'的模块

时间:2019-08-06 00:44:35

标签: python tensorflow

花了几天的时间进行故障排除。这是我正在学习的课程的完整安装指南

https://github.com/reigngt09/Pose-Estimation/tree/master/1.%20Pose%20Estimation%20Installation%20Guide

当我尝试在cmd中输入以下内容

$python run_webcam.py --camera video.mp4

我收到以下错误

ModuleNotFoundError: No module named 'tensorflow.contrib'

违规行是

import tensorflow.contrib.tensorrt as trt

这是我的设置规格

  

Windows 10

     

Python 3.6.8

     

CUDA 10.0

     

cuDNN v 7.6.2

     

Tensorflow(gpu)1.14.0

     

GeForce GTX 960M

     

驱动程序版本431.60

     

Intel Core i7-6700HQ 2.6 GHz *

感谢任何反馈或故障排除步骤!

1 个答案:

答案 0 :(得分:0)

您可以尝试通过以下方式导入:

from tensorflow.python.compiler.tensorrt import trt_convert as trt

https://docs.nvidia.com/deeplearning/frameworks/tf-trt-user-guide/index.html#usage-example