【问题标题】:Loading tensorflow library into C++将张量流库加载到 C++ 中
【发布时间】:2017-12-15 04:43:32
【问题描述】:

我在 python 中有一个来自 Keras 的训练模型,我想使用 tensorflow 1.2 在 C++ 中进行预测。我似乎无法将 tensorflow 库放入我的 c++ 程序中。任何人都可以带我完成执行此操作的步骤? ii)如果需要,我是否需要使用 bazel 使用 bazel 的步骤是什么? 谢谢:)

【问题讨论】:

    标签: c++ tensorflow


    【解决方案1】:

    我已经通过 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 并选择 tensorflow 选项,启用“共享库”并启用 GPU 支持。

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

    会有一些错误,但很容易修复。如果您希望了解更多详细信息,请发表评论,很高兴为您提供帮助。

    【讨论】:

      猜你喜欢
      • 2018-09-11
      • 2021-11-27
      • 2023-03-28
      • 1970-01-01
      • 1970-01-01
      • 2017-03-03
      • 1970-01-01
      • 2017-11-05
      相关资源
      最近更新 更多