将tensorflow库加载到C ++中

时间:2017-07-11 08:22:16

标签: c++ tensorflow

我有一个训练有素的模型来自Keras的python,我想使用tensorflow 1.2在C ++中进行预测。我似乎无法将tensorflow库放入我的c ++程序中。谁能带我完成这个步骤? ii)如果是的话,我是否需要使用bazel使用bazel的步骤是什么? 谢谢:))

1 个答案:

答案 0 :(得分:0)

我已经通过CMake和Visual Studio成功构建了tensorflow-gpu动态库。现在能够用inception_V3和RCNN训练的模型预测。

我使用过Visual Studio 2015,CUDA 9.0,cudNN 7.0,Python 3.5。

以下简要介绍我的步骤:

  1. git clone -b r1.7 --single-branch https://github.com/tensorflow/tensorflow.git

  2. 使用CMake GUI并选择张量流选项,启用“共享库”并启用-gpu-support。

  3. 使用发布模式在Visual Studio中构建tensorflow.dlltensorflow.lib

  4. 会有一些错误,但很容易修复。如果您希望了解更多详情,请发表评论,很乐意提供帮助。