记 Windows 10 编译 tensorflow

作者 : 开心源码 本文共1862个字,预计阅读时间需要5分钟 发布时间: 2022-05-12 共100人阅读

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 源码 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日

说明
1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长!
2. 分享目的仅供大家学习和交流,您必须在下载后24小时内删除!
3. 不得使用于非法商业用途,不得违反国家法律。否则后果自负!
4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解!
5. 如有链接无法下载、失效或广告,请联系管理员处理!
6. 本站资源售价只是摆设,本站源码仅提供给会员学习使用!
7. 如遇到加密压缩包,请使用360解压,如遇到无法解压的请联系管理员
开心源码网 » 记 Windows 10 编译 tensorflow

发表回复