首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >在cmake中使用预编译的tensorflow

在cmake中使用预编译的tensorflow
EN

Stack Overflow用户
提问于 2016-06-28 12:58:04
回答 2查看 6K关注 0票数 11

我在CLion中设置了一个使用CMake项目。我正在使用各种第三方库,并希望还集成Tensorflow。

我已经尝试过bazel将Tensorflow编译成一个共享库libtensorflow.so,但是仍然有相当多的依赖项(例如,当前的protobuf版本,一旦我这样做了,还有更多的依赖项)需要修复。

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

Python中的Tensorflow很适合我。

EN

回答 2

Stack Overflow用户

发布于 2016-08-01 16:02:47

我知道这个答案已经很晚了,但我遇到了你的确切问题,并能够解决它。我创建了一个存储库这里,它描述了如何实现您想要的结果。要点是:

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

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

票数 6
EN

Stack Overflow用户

发布于 2018-12-18 15:52:56

如果你在MacOS上,用自制的,CMake和pkg_config,这很容易。

首先使用brew获取Tensorflow:

代码语言:javascript
复制
brew install libtensorflow

然后在CMakeLists.txt

代码语言:javascript
复制
cmake_minimum_required(VERSION 3.10)
project(tf-inference)

find_package(PkgConfig)
pkg_check_modules(TensorFlow REQUIRED tensorflow)

link_directories(${TensorFlow_LIBRARY_DIRS})
include_directories(${TensorFlow_INCLUDE_DIRS})
add_compile_definitions(${TensorFlow_CFLAGS_OTHER})

add_executable(tf-inference inference.cpp)
target_link_libraries(tf-inference ${TensorFlow_LIBRARIES})
票数 5
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/38076914

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档