0x00 前言

第一次在Windows下做机器学习开发(PS: 真的第一次用Windows做机器学习,之前都是OS X 或者Linux下做的,吐槽Windows,真的好难用,就是好难用,不服你顺着网线来打我啊!哼~)。上司说我写的模型怎么运行那么慢啊?这怪我喽?! 一个8G+i5 4210U的笔记本,我也很无奈啊。之后就想到了一个问题,我用的tensorflow是直接pip安装的,官方为了保证一定的兼容性,仅使用了CPU SSE指令集做优化,并没有做 AVX/AVX2 指令集优化,然而,这怎么对得起我这i5 4210U的处理器呢?所以,接下来,我们开始在Windows下编译tensorflow。

0x01 工具准备

  1. Python 3.5/3.6 64位版本 下载链接
  2. swigwin 3.0.12版本  下载链接
  3. cmake 3.12.0 rc3版本 下载链接 ⚠这个版本的工具发生改变,如安装低版本的cmake请先卸载后再进行安装
  4. git 2.18.0版本 下载链接
  5. VS2017 本次编译仅供支持AVX2指令集
  6. 最重要最重要的 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 注意事项

  1. 安装swigwin 时仅需要下载之后解压即可,在系统的PATH中添加该文件的文件夹路径

  2. 安装完成python之后,我们需要先安装一次pip的tensorflow以获得编译时所依赖的其他库,命令pip install tensorflow

  3. 编译时必须链接网络,其需要从github等网站上自动下载所依赖的文件

  4. 编译时间较长,请耐心等待

  5. 编译 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日