系统:MacOS sierra 10.12.6
tf版本:1.2.1
前言
最近在安装tensorflow的时候,遇到了下面的问题:
实际上就是因为我们直接用官方编译好的,并不能完美的与本地的指令集兼容,有点强迫症,所以决定从源码进行编译。
1、环境准备
基本的环境主要包括这几步:
安装bazel
安装python依赖
NVIDIA 的包(如果有显卡)
mac并没有NVIDIA显卡,所以这一步省略。
基本的科学环境使用的是anaconda,如果不太熟的话,参见参考文献中的anaconda博客,网上也有很多相关教程。
1.1 安装brew
brew是一个非常好用的mac的包管理器,能够自动的帮助我们解决依赖问题,安装卸载都非常方便。
/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
如果还没有git的话,建议使用brew安装git。
brew install git
1.2 代理设置
经常能到大家抱怨使用git或者brew的时候,速度很慢,实际上可以使用代理来加快速度,如果已经使用shadowsocks搭建了代理服务器,那就好了,只需要简单的配置即可。
需要说明的是,在mac下,新版本的系统开启了SIP(System Integrity Protection),所以proxychains不好用,除非关闭,不过这样还是有点麻烦,简单的来说,直接用下面的配置简单快捷。
在环境变量中导出下面的变量:
export ALL_PROXY=socks5://127.0.0.1:1086
这样的话,在命令行中就可以使用代理了,经过测试,git最高速度差不多到4MB左右。
1.3 安装bazel
使用brew安装bazel很简单:
brew install bazel
➜ ~ bazel version
Build label: 0.5.3-homebrew
Build target: bazel-out/darwin_x86_64-opt/bin/src/main/java/com/google/devtools/build/lib/bazel/BazelServer_deploy.jar
Build time: Sun Aug 6 12:54:10 2017 (1502024050)
Build timestamp: 1502024050
Build timestamp as int: 1502024050
请参考参考文献中的bazel官方文档。
1.4 安装python依赖
首先确认我们已经准备好了anaconda的环境,然后创建一个环境,我的环境是python3.6.2版本,环境名为py36,首先使用下面的命令激活环境:
source activate py36
然后使用conda命令安装相关依赖:
conda install six numpy wheel
在编译的时候,需要调用这些包,所以,建议是先激活环境,防止出错。
1.5 tensorflow源码下载
使用git下载tensorflow的源码:
git clone https://github.com/tensorflow/tensorflow
cd tensorflow
ls
ACKNOWLEDGMENTS AUTHORS CONTRIBUTING.md LICENSE RELEASE.md bazel-bin bazel-out bazel-testlogs configure tensorflow tools
ADOPTERS.md BUILD ISSUE_TEMPLATE.md README.md WORKSPACE bazel-genfiles bazel-tensorflow bower.BUILD models.BUILD third_party util
然后看一下tag:
git tag
0.12.0-rc0
0.12.0-rc1
0.12.1
0.5.0
0.6.0
v0.10.0
v0.10.0rc0
v0.11.0
v0.11.0rc0
v0.11.0rc1
v0.11.0rc2
v0.12.0
v0.6.0
v0.7.0
v0.7.1
v0.8.0
v0.8.0rc0
v0.9.0
v0.9.0rc0
v1.0.0
v1.0.0-alpha
v1.0.0-rc0
v1.0.0-rc1
v1.0.0-rc2
v1.0.1
v1.1.0
v1.1.0-rc0
v1.1.0-rc1
v1.1.0-rc2
v1.2.0
v1.2.0-rc0
v1.2.0-rc1
v1.2.0-rc2
v1.2.1
v1.3.0
v1.3.0-rc0
v1.3.0-rc1
v1.3.0-rc2
选择我们想要的版本,切换过去:
git checkout v1.2.1
这样我们就把环境准备好了。
2、编译
2.1 配置
在tensorflow的目录中,运行下面的命令:
./configure
会出现一些提示,包括python的库的位置,根据实际情况填写,参考参考文献中的官方文档。
2.2 编译
编译,创建pip包:
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
这个是cpu的版本,没有GPU。
大约20多分钟以后,终于编译完成。
接下来打包成wheel包,以便安装:
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
然后,我们在/tmp/tensorflow_pkg
目录下就能看到这样的文件:
tensorflow-1.2.1-cp36-cp36m-macosx_10_7_x86_64.whl
然后使用下面的命令,安装到当前的anaconda的环境:
pip install /tmp/tensorflow_pkg/tensorflow-1.2.1-cp36-cp36m-macosx_10_7_x86_64.whl
然后,再也不会有烦人的提示了!
3、参考文献
[官方文档]:https://www.tensorflow.org/install/install_sources
[anaconda博客]:http://www.jianshu.com/p/2f3be7781451
[brew官网]:https://brew.sh/index_zh-cn.html
[homebrew替换中科大的源]:https://lug.ustc.edu.cn/wiki/mirrors/help/brew.git
[中科大homebrew源]:https://lug.ustc.edu.cn/wiki/mirrors/help/homebrew-bottles
[bazel官方文档]:https://docs.bazel.build/versions/master/install-os-x.html