在Windows平台上使用Python的TensorRT

Question

4

我在Windows上使用TensorRT for python时遇到了问题。

我想在Windows上使用openpose，但它需要TensorRT for python。当我尝试执行run_webcam.py（openpose代码）文件时，出现以下错误：

ModuleNotFoundError: No module named "tensorflow.contrib.tensorrt"

我安装了tensorflow（最新版本），但它不包含tensorrt类。通过查找解决方法，我在nvidia的网站上找到了以下信息https://devtalk.nvidia.com/default/topic/1044414/tensorrt/tensorrt-python-and-windows/

如果windows不支持Python的TensorRT，那么OpenPose在Windows上是如何工作的？

非常感谢任何提示或帮助。

- Edna Salazar

嗨，@Edna Salazar。据我所知，官方OpenPose不使用TensorFlow和TensorRT，而是使用Caffe，可以在Windows上运行。您是使用官方的OpenPose还是众多第三方克隆版本之一？ - hbaderts

2个回答

0

现在在Windows下有官方支持的TensorRT Python API。请查看以下问题中NVIDIA主持人的评论：

为了检查不同版本的TensorRT、Tensorflow、cuDNN和其他库之间的兼容性，请访问以下链接： https://docs.nvidia.com/deeplearning/tensorrt/support-matrix/index.html

- jd95

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- JVGD · Accepted Answer

据我所知，根据官方TensorRT文档（链接），TensorRT Python API在Windows上不受支持：

TensorRT的Windows zip包不提供Python支持。未来可能会支持Python。

在Windows系统中进行推理时，我也遇到了同样的问题。我正在使用TRT进行Python推理，而在Linux中它非常完美...... 我们已经发布了TRT 7.0版本，但仍没有支持Windows上的Python API的计划。