【问题标题】:Creating a build environment to build Tensorflow apps with cmake使用 cmake 创建构建环境以构建 TensorFlow 应用程序
【发布时间】:2018-07-31 06:49:19
【问题描述】:

我想知道是否有使用 cmake 为应用程序构建 tensorflow 和 tensor 的明确方法。我按照https://github.com/cjweeks/tensorflow-cmake 的说明进行操作,但没有成功,最终不得不手动构建 Eigen 和 Protobuf,然后将相关的头文件复制到由 Tensorflow 的 Bazel 构建创建的头文件树中。

【问题讨论】:

    标签: tensorflow cmake


    【解决方案1】:

    我刚刚使用 CMake、VS2017 和 CUDA 9.2 构建了 TF,但不得不进行两项手动更改:

    1. 补丁Half.h in Eigen

    2. 在主 CMakeLists.txt 中将 CUDA 版本从“9.0”更改为“9.2”。

    构建必须是单线程的,否则 VS 会耗尽堆(在我的 16GB 笔记本电脑上)。这需要一段时间,一个项目失败了,但构建了足够的库来运行我想要的所有示例。

    与 Bazel 相比,CMake 构建的另一个问题是前者会重建一堆项目(涉及 protobuf 生成的文件),即使没有任何变化。 Bazel 更智能,只编译更改的代码,然后将所有目标文件静态链接到一个可执行文件中,这仍然比 CMake 构建更快。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2019-01-30
      • 2013-04-09
      • 2018-01-15
      • 2013-08-22
      • 1970-01-01
      • 1970-01-01
      • 2010-10-17
      • 1970-01-01
      相关资源
      最近更新 更多