我有一个训练有素的模型来自Keras的python,我想使用tensorflow 1.2在C ++中进行预测。我似乎无法将tensorflow库放入我的c ++程序中。谁能带我完成这个步骤? ii)如果是的话,我是否需要使用bazel使用bazel的步骤是什么? 谢谢:))
答案 0 :(得分:0)
我已经通过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并选择张量流选项,启用“共享库”并启用-gpu-support。
使用发布模式在Visual Studio中构建tensorflow.dll
和tensorflow.lib
。
会有一些错误,但很容易修复。如果您希望了解更多详情,请发表评论,很乐意提供帮助。