gpt4 book ai didi

c++ - 将预编译的tensorflow与cmake一起使用

转载 作者:可可西里 更新时间:2023-11-01 16:09:42 26 4
gpt4 key购买 nike

我在 CLion 中设置了一个使用 CMake 的 C++ 项目。我正在使用各种第 3 方库,并且还想集成 Tensorflow

我试过 bazelTensorflow 编译成一个共享库 libtensorflow.so 哪种方法可行,但是仍然有很多我必须修复的依赖项(例如,当前的 protobuf 版本,一旦我这样做了,就会有更多)。

有没有办法使用标准的 Tensorflow git 存储库并以某种方式链接为 python 使用而预编译的库?或者有其他方便的方法吗?

Python 中的 Tensorflow 非常适合我。

最佳答案

我知道这个答案已经很晚了,但我遇到了你的确切问题并且能够解决它。我创建了一个存储库 here描述了如何准确地完成你想要的。要点是:

  • 将构建规则添加到包含所有必需的 C++ 元素的 TensorFlow 存储库。
  • 使用 Bazel 构建共享库并将所有 header 复制到 /usr/local
  • 安装特定版本的 Protobuf 和 Eigen(这是通过脚本自动完成的)或将它们添加为外部依赖项。
  • 使用给定的文件配置您的 CMake 项目。

如果您有任何疑问或问题,请随时与我联系。

关于c++ - 将预编译的tensorflow与cmake一起使用,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38076914/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com