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-and-windows/上找到了以下信息

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

非常感谢任何提示或帮助

Answer 1

就我而言，根据 official TensorRT documentation，Windows 不支持 TensorRT python API。 :

The Windows zip package for TensorRT does not provide Python support. Python may be supported in the future

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