ubuntu16.04下TensorFlow软件库平台

写在前面:

前段时间将《概率论》和《线性代数》的一些基础知识重新看了下,准备去接触机器学习这个方向;事实上,一年来已经断断续续看过不少相关内容,只是一直未成体系,因此未敢动笔,直到最近才感觉时机差不多,后面会尝试着慢慢写来。

由于某些大家已知的原因,导致 tensorflow.org 网站国内无法访问;在此,强烈建议访问网址:https://tensorflow.google.cn/

本篇从 TensorFlow 的安装开始,将介绍在 Ubuntu 下安装 TensorFlow。

1. 安装 pip

安装 pip 和 virtualenv 命令(Python3为例)

sudo apt-get install python3-pip python3-dev python-virtualenv

virtualenv 用于创建独立的 Python 环境,多个 Python 相互独立,互不影响,它能够:

  • 在没有权限的情况下安装新套件

  • 不同应用可以使用不同的套件版本

  • 套件升级不影响其他应用

2. 创建虚拟环境

在桌面打开终端,创建一个虚拟环境 env 。

virtualenv env

图片

3. 启动虚拟环境

进入env目录,启动虚拟环境。

cd env
source ./bin/activate

图片

4. 更改pip镜像源

修改pip镜像源为国内,具体列表,查看网址:https://www.pypi-mirrors.org/

4.1 Linux

linux下,修改 ~/.pip/pip.conf (没有就创建一个),添加源如下:

[global]
index-url = https://pypi.doubanio.com/simple/

4.2 windows

windows下,参考这篇文章《Win7下cuda8.0深度学习开发平台》

5. 其它工具

此处建议安装下几个科学计算工具,如下:

pip3 install matplotlib
pip3 install numpy
pip3 install scipy
pip3 install scikit-image

图片

6. 安装 TensorFlow

安装最新 TensorFlow 版本。

pip3 install tensorflow        #CPU版本(二选一)
pip3 install tensorflow-gpu    #GPU版本(二选一)

图片

7. 测试安装是否成功

7.1 测试源码

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

7.2 测试过程

图片

8. 卸载 tensorflow

如果需要卸载的话,使用如下:

pip3 uninstall tensorflow

9. 退出env环境

deactivate

10. 删除虚拟环境

如果需要删除虚拟环境,如下:

rm -r env

注意:

如果在安装的过程中出现以下错误:

File "/usr/lib/python3.5/tkinter/__init__.py", line 38, in <module>

 raise ImportError(str(msg) + ', please install the python3-tk package')

ImportError: No module named '_tkinter', please install the python3-tk package

解决方法:

sudo apt-get isntall python3-tk

参考内容:

https://tensorflow.google.cn/install/install_linux

http://tensorlayer.readthedocs.io/en/latest/user/installation.html

http://tensorlayer.readthedocs.io/en/latest/user/tutorial.html#run-the-mnist-example