记 Windows 10 编译 tensorflow
0x00 前言
第一次在Windows下做机器学习开发(PS: 真的第一次用Windows做机器学习,之前都是OS X 或者Linux下做的,吐槽Windows,真的好难用,就是好难用,不服你顺着网线来打我啊!哼~)。上司说我写的模型怎么运行那么慢啊?这怪我喽?! 一个8G+i5 4210U的笔记本,我也很无奈啊。之后就想到了一个问题,我用的tensorflow是直接pip安装的,官方为了保证一定的兼容性,仅使用了CPU SSE指令集做优化,并没有做 AVX/AVX2 指令集优化,然而,这怎么对得起我这i5 4210U的处理器呢?所以,接下来,我们开始在Windows下编译tensorflow。
0x01 工具准备
- Python 3.5/3.6 64位版本 下载链接
- swigwin 3.0.12版本 下载链接
- cmake 3.12.0 rc3版本 下载链接 ⚠这个版本的工具发生改变,如安装低版本的cmake请先卸载后再进行安装
- git 2.18.0版本 下载链接
- VS2017 本次编译仅供支持AVX2指令集
- 最重要最重要的 tensorflow 源码 https://github.com/tensorflow/tensorflow
注:软件版本不必一定相同,一般来说只要能用即可,VS务必安装2017版本,2015版本不支持AVX2指令集。
如果不明白上面的软件是什么或者不会下载安装上述软件,请你点个赞之后关闭本网页,下述的内容对你可能会产生严重不适。
Warning
0x02 开始编译
打开克隆好的 tensorflow 仓库文件夹,打开CMake程式。
CMake GUI版本 在仓库的 tensorflow\contrib\cmake 文件夹下新建一个文件夹,命名 build (这个可以随便命名,这里是编译的地方,其记住这两个路径)。 填写路径并开始configure(配置)
这个使用默认即可
默认选择即可
如果没问题,请跳过下面一段,并点击Generate开始生成项目吧。
如果出了问题,而且日志如我图中所示
SWIG未找到
那么就手动设置这个路径到你解压的SWIG文件夹路径里就可以了。 手动设置这个路径
之后再点击Configure,看到Configuring done。
解决上面的那个问题了点击Generate开始生成项目吧。
多了这么多的项目文件打开VS 2017的开发人员命令提示符,
务必点击使用这个的说在终端中打开到build的文件夹里,因为我就只需要pip安装编译好的包,故我只需要编译这个即可,输入MSBuild /p:Configuration=Release tf_python_build_pip_package.vcxproj
就耐心等待吧。
编译完成之后我们就可以在build路径下的 tf_python\dist 文件夹中看到一个whl结尾的文件,接下来就尽情使用吧。
0x03 注意事项
-
安装swigwin 时仅需要下载之后解压即可,在系统的PATH中添加该文件的文件夹路径
-
安装完成python之后,我们需要先安装一次pip的tensorflow以获得编译时所依赖的其他库,命令
pip install tensorflow
-
编译时必须链接网络,其需要从github等网站上自动下载所依赖的文件
-
编译时间较长,请耐心等待
-
编译 re2.vcxproj 时出错
*warning C4819: The file contains a character that cannot be represented in the current code page (936). Save the file in Unicode format to prevent data loss*
代码图
以及
Error C2001: Newline in constant
错误原因:此错误和编译平台的 windows 操作系统的locale设为中文有关,英文的 windows 没有这个错误
解决办法: 修改 CMakeCache.txt,让 MSBuild 编译这个项目时,强制按照 utf-8 编码文件的格式来解析文件
进入 build\re2\src\re2 目录
用文本编辑器打开 CMakeCache.txt,找到以下2行
在 CMAKE_CXX_FLAGS_RELEASE 这里添加蓝色部分,修改为
CMAKE_CXX_FLAGS_RELEASE:STRING=/MD /O2 /Ob2 /DNDEBUG /source-charset:utf-8
0x04 结束语
安装之后速度提升了一些,但是还是要吐槽这笔记本,没显卡,内存小,低处理器,老板还嫌我的速度慢,😕哼~
如果帮到你了的话,就点个赞打个赏什么的吧,我...我才不会感谢你呢。哼~~
By 土豆豆,2018年7月17日