【发布时间】:2018-07-31 06:49:19
【问题描述】:
我想知道是否有使用 cmake 为应用程序构建 tensorflow 和 tensor 的明确方法。我按照https://github.com/cjweeks/tensorflow-cmake 的说明进行操作,但没有成功,最终不得不手动构建 Eigen 和 Protobuf,然后将相关的头文件复制到由 Tensorflow 的 Bazel 构建创建的头文件树中。
【问题讨论】:
标签: tensorflow cmake
我想知道是否有使用 cmake 为应用程序构建 tensorflow 和 tensor 的明确方法。我按照https://github.com/cjweeks/tensorflow-cmake 的说明进行操作,但没有成功,最终不得不手动构建 Eigen 和 Protobuf,然后将相关的头文件复制到由 Tensorflow 的 Bazel 构建创建的头文件树中。
【问题讨论】:
标签: tensorflow cmake
我刚刚使用 CMake、VS2017 和 CUDA 9.2 构建了 TF,但不得不进行两项手动更改:
补丁Half.h in Eigen
在主 CMakeLists.txt 中将 CUDA 版本从“9.0”更改为“9.2”。
构建必须是单线程的,否则 VS 会耗尽堆(在我的 16GB 笔记本电脑上)。这需要一段时间,一个项目失败了,但构建了足够的库来运行我想要的所有示例。
与 Bazel 相比,CMake 构建的另一个问题是前者会重建一堆项目(涉及 protobuf 生成的文件),即使没有任何变化。 Bazel 更智能,只编译更改的代码,然后将所有目标文件静态链接到一个可执行文件中,这仍然比 CMake 构建更快。
【讨论】: