python - 用于 windows 上的 python 的 TensorRT

Question

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

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

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

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

如果 windows 不支持 python 的 TensorRT，OpenPose 如何在 windows 上工作？？？？？？

非常感谢任何提示或帮助

Answer 1

就我而言，根据 TensorRT官方文档，Windows 不支持 TensorRT python API ：

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

对于 Windows 系统中的推理，我遇到了同样的问题。我在 python 中使用 TRT 进行推理，它就像一个魅力......在 Linux 中。我们已经在 TRT 7.0 版本中，仍然没有计划在 Windows 中支持 python API。