首页 > 解决方案 > Tensorflow > ModuleNotFoundError:没有名为“tensorflow.contrib”的模块

问题描述

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

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

这是我的设置规格

视窗 10

Python 3.6.8

CUDA 10.0

cuDNN v 7.6.2

张量流(GPU)1.14.0

GeForce GTX 960M

驱动程序版本 431.60

英特尔酷睿 i7-6700HQ 2.6 GHz*

任何反馈或故障排除步骤表示赞赏!

标签: pythontensorflow

解决方案


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

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

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


推荐阅读