TensorFlow学习笔记（三）—— linux安装TensorFlow

亦凉 2021-09-27 09:58 716阅读 0赞

参照官方文档https://www.tensorflow.org/get\_started/os\_setup.html  
我们可以使用pip来安装tensorflow  
因为依赖python,且有python2与python3之分，所以先安装virtualenv  
可方便得在python2.7与python3之前动态切换  
sudo pip install virtualenv  
  
  
激活环境  
virtualenv --system-site-packages ~/tensor  
cd ~/tensor/  
bash的话  
source bin/activate  
如果是csh则source bin/activate.csh  
注意，这一步每次要执行tensorflow时都要执行的  
  
  
这里pip与pip3有区别的,我用的python3的版本  
注意，如果是gpu的还要安装CUDA9.0工具包,与 CUDA 工具包 9.0 相关联的 NVIDIA 驱动程序,cuDNN v7.0, libcupti-dev 库  
sudo pip3 install --upgrade tensorflow-gpu  
如果不带gpu则是  
sudo pip3 install --upgrade tensorflow  
如果是python2则把上面的pip3改成pip  
  
  
退出环境  
deactivate  
  
  
要缷载掉tensorflow, 这后面的tensorflow-gpu要对应前面装的，如果是gpu的后面带-gpu  
sudo pip3 uninstall tensorflow-gpu  
  
  
源码编译tensorflow  
安装java-1\_8\_0-openjdk  
安装bazel 这个可以从源码或者直接系统安装  
如果要编译bazel则要安装java 8 jdk,可以是openjdk-8 其他版本的不行  
源码方法：  
从  
https://github.com/bazelbuild/bazel/releases  
下载最新的版本，我下的是bazel-0.13.0-dist.zip  
mkdir bazel  
cd bazel  
unzip -x bazel-0.13.0-dist.zip  
bash ./compile.sh  
可以参考官方的https://docs.bazel.build/versions/master/install.html  
  
  
git clone https://github.com/tensorflow/tensorflow   
cd tensorflow  
./configure  
bazel build --config=opt --config=cuda //tensorflow/tools/pip\_package:build\_pip\_package  
bazel-bin/tensorflow/tools/pip\_package/build\_pip\_package /tmp/tensorflow\_pkg（将在/tmp/tensorflow\_pkg目录下面生成对应的\*.whl）  
sudo pip install /tmp/tensorflow\_pkg/tensorflow-YOUR-VERSION.whl  
上面的--config=cuda 去掉就是编译CPU版

这部分是配置tensorflow，然后再生成whl安装tensorflow。   
个人理解：直接pip安装就是安装官网提供的已经配置好的whl包，而源码安装就是利用bazel编译后，生成whl包，再进行安装。  
  
检验是否安装正常：  
(tensor) zm@linux-kl9l:~/tensor> python3  
Python 3.6.5 (default, Mar 31 2018, 19:45:04) \[GCC\] on linux  
Type "help", "copyright", "credits" or "license" for more information.  
>>> import tensorflow as tf  
>>>   
  
  
其他参考  
https://blog.csdn.net/wenanshi/article/details/73484688  
https://blog.csdn.net/u012436149/article/details/53312212  
http://vividfree.github.io/%E5%B7%A5%E5%85%B7%E5%AE%89%E8%A3%85/2017/04/13/install-tensorflow-on-linux-from-sources  
http://wiki.jikexueyuan.com/project/tensorflow-zh/get\_started/os\_setup.html

**备注：**

**1.TensorFlow 1.2版本之后，需要Python 3.5以上的版本**

**2.基于pip安装方式，需要的XXXX.whl，需要翻墙，或者去下面地址自行下载：**

[https://github.com/mind/wheels][https_github.com_mind_wheels]

[https_github.com_mind_wheels]: https://github.com/mind/wheels