Tensorflow工程实践: tensorflow1.8 + vs2017 + cmake 编译生成c++ 库包

最新推荐文章于 2023-03-04 13:54:13 发布

cfqcfqcfqcfqcfq

最新推荐文章于 2023-03-04 13:54:13 发布

阅读量2.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow

本文链接：https://blog.youkuaiyun.com/cfqcfqcfqcfqcfq/article/details/90582154

本文详细记录了使用Tensorflow1.8、Visual Studio 2017和CMake在Windows环境下编译生成C++库包的过程，包括环境配置、CMake构建、VS编译命令、解决常见错误及调用实例。编译过程中遇到的问题，如gRPC支持关闭、Python debug库的选择、内存不足错误和re2测试禁用等，都有详尽的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

因为要做工程化，所以最近用了一些时间来研究c++版本的tensorflow 当然官网没有提供现成的库包，只能自己根据原码编译。网上翻了一遍，发现资料很多但都相对较为零散，这边记录一下自己编译库包(Release和Debug版本)的流程以及自己踩的坑。

个人尝试了很多个版本，但相对只有1.8版本通过编译，1.10版本总有rpc相关错误(设置了OFF，任报错)，在往上的高版本，cmake生成vs工程都困难重重，就没有继续研究下去。

个人电脑环境:

win10

vs2017

cmake3.8.2(3以上版本都可)

Swigwin-3.0.12

tensorflow-r1.8

python3.6(Anaconda)

网络要求:

编译时候会从git和一些国外网站在线下载一些依赖库包，所以提取要准备好自己的网络(各位应该都懂。。)，否则一直处于time out状态.

Cmake构建：

在tensorflow-r1.8\tensorflow\contrib\cmake下面新建一个build文件夹，作为vs工程的生成目录.

修改cmakeList文件：

1、招到if (tensorflow_OPTIMIZE_FOR_NATIVE_ARCH)那一段代码，用下面的替换:

if (tensorflow_OPTIMIZE_FOR_NATIVE_ARCH)
  include(CheckCXXCompilerFlag)
  CHECK_CXX_COMPILER_FLAG("-march=native" COMPILER_OPT_ARCH_NATIVE_SUPPORTED)
  if