花了几天的时间进行故障排除。这是我正在学习的课程的完整安装指南
当我尝试在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 *
感谢任何反馈或故障排除步骤!
答案 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