因为要做工程化,所以最近用了一些时间来研究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)
in