正如标题所说，我正在尝试使用 Nvidia 的 tensor-rt 库在 c++ 代码中使用 Python 中实现的预训练 tensorflow 网络。我知道我需要从“冻结图”创建一个 .pb 文件（我不知道那是什么意思），而不是在 c++ 中以某种方式使用 tensor-rt API。有人可以帮我吗？
user6531747
问问题
233 次
2 回答
0
如果您打算使用TF-TRT，这里是它的文档：https ://docs.nvidia.com/deeplearning/dgx/integrate-tf-trt/index.html
这里也有一些例子可以帮助你入门：https ://github.com/tensorflow/tensorrt
于 2018-12-11T05:43:56.360 回答
0
这里有一个关于如何做到这一点的指南，它是由 Nvidia 的人制作的，如果你安装了 tensorrt，你将在这个方向上获得关于 TF 部署的示例代码：
/usr/src/tensorrt/samples/sampleUffMNIST
并且一如既往地查看 TensorRT 文档https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html
那应该让你开始！祝你好运！
于 2018-12-07T10:00:05.793 回答