美文网首页技术文技术干货
记 Windows 10 编译 tensorflow

记 Windows 10 编译 tensorflow

作者: 土豆豆一只 | 来源:发表于2018-07-17 14:22 被阅读7次

    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日

    相关文章

      网友评论

        本文标题:记 Windows 10 编译 tensorflow

        本文链接:https://www.haomeiwen.com/subject/kfnwpftx.html