TensorFlow的2种安装方法---pip安装和源码安装

本篇学习笔记是在 macOS 系统上进行的演示,python 版本为 2.7.10,如果大家想学习在 windowslinux 等系统上的安装步骤,可以参考一下官方的安装 TensorFlow 的方法:https://www.tensorflow.org/install/ ,个人建议在虚拟环境下安装 TensorFlow。下面介绍 TensorFlow 的2种安装方法。

通过「pip」直接安装

安装「pip」

如果已经安装,跳过此步

1
sudo easy_install pip

查看 pip 是否安装成功

1
pip —version

安装「tensorflow」

安装CPU版本:

1
pip install tensorflow

安装GPU版本:

1
pip install tensorflow-gpu

注:默认安装都是最新版本,如需要安装指定版本,需要加上版本号
例如安装1.0.0版本
安装CPU版本:

1
pip install tensorflow==1.0.0

安装GPU版本:

1
pip install tensorflow-gpu==1.0.0

通过源码安装

首先Mac下内置有gcc,可以通过终端命令gcc -v 查看:

其中的4.2.1就是gcc的版本,或者大家也可以直接输入终端命令python进行查看

里面有一个GCC 4.2.1,感觉比上面那个命令更直观一些。
然后就是要安装Google自家的编译工具bazel了,mac上最好的办法就是直接使用brew安装了:\
brew install bazel
安装好之后就要下载官方的TensorFlow源码了:
git clone https://github.com/tensorflow/tensorflow
下载完成后会在你的用户名的目录下生成一个tensorflow的文件夹,使用
cd tensorflow
进入这个文件夹,然后输入:
./configure
进行配置,通常情况下没有特殊的要求,一般全部都选择n,需要输入path的地方敲回车,使用默写path

配置完成后就是编译过程了,输入终端命令:
bazel build –config=opt //tensorflow/tools/pip_package:build_pip_package
经过漫长的编译后,任务就基本上完成了,接下来就是输入下面的命令来生成pip安装包了:
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
接下来安装这个wheel。如果之前pip装过tf的话,最好先卸载掉:
pip uninstall tensorflow
再安装
pip install /tmp/tensorflow_pkg/tensorflow-1.3.0-cp27-cp27m-macosx_10_12_intel.whl
安装成功
Installing collected packages: tensorflow
Successfully installed tensorflow-1.2.0
验证安装
遇到的问题:
如果错误是bazel相关的,请卸载bazel重新安装。

-------------本文结束感谢您的阅读-------------
坚持整理学习笔记,您的支持将鼓励我继续整理下去!