TensorFlow学习笔记(三)—— linux安装TensorFlow

参照官方文档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

发布了48 篇原创文章 · 获赞 121 · 访问量 68万+

猜你喜欢

转载自blog.csdn.net/guoyunfei123/article/details/82429193