【发布时间】:2017-12-15 04:43:32
【问题描述】:
我在 python 中有一个来自 Keras 的训练模型,我想使用 tensorflow 1.2 在 C++ 中进行预测。我似乎无法将 tensorflow 库放入我的 c++ 程序中。任何人都可以带我完成执行此操作的步骤? ii)如果需要,我是否需要使用 bazel 使用 bazel 的步骤是什么? 谢谢:)
【问题讨论】:
标签: c++ tensorflow
我在 python 中有一个来自 Keras 的训练模型,我想使用 tensorflow 1.2 在 C++ 中进行预测。我似乎无法将 tensorflow 库放入我的 c++ 程序中。任何人都可以带我完成执行此操作的步骤? ii)如果需要,我是否需要使用 bazel 使用 bazel 的步骤是什么? 谢谢:)
【问题讨论】:
标签: c++ tensorflow
我已经通过 CMake 和 Visual Studio 成功构建了一个 tensorflow-gpu 动态库。现在可以使用 inception_V3 和 RCNN 训练的模型进行预测。
我使用过 Visual Studio 2015、CUDA 9.0、cudNN 7.0、Python 3.5。
以下是我的步骤的简要介绍:
git clone -b r1.7 --single-branch https://github.com/tensorflow/tensorflow.git
使用 CMake GUI 并选择 tensorflow 选项,启用“共享库”并启用 GPU 支持。
在 Visual Studio 中使用发布模式构建 tensorflow.dll 和 tensorflow.lib。
会有一些错误,但很容易修复。如果您希望了解更多详细信息,请发表评论,很高兴为您提供帮助。
【讨论】: