Ubuntu利用virtualenv安装TensorFlow

来自CloudWiki
跳转至: 导航搜索

软硬件环境

  • 操作系统:Ubuntu 16.04 LTS
  • 硬件支持:TensorFlow with CPU support only.
这个文档是针对在没有NVIDIA GPU的机器上安装TensorFlow来写的。安装过程非常快(5-10分钟),所以即使你有一个NVIDIA GPU,也推荐先按这种方法安装。

安装方法

  • TensorFlow官网给出了4种安装方法,我们这里采用最简单的virtualenv安装
  • Virtualenv用来为一个应用创建一套“隔离”的Python运行环境。在实际开发中每个应用可能需要各自拥有一套“独立”的Python运行环境,virtualenv能够方便的为每个应用各自提供一个“独立”的Python运行环境,并且相互之间互不打扰。Virtualenv也是安装TensorFlow的神器。

具体步骤

  • 执行如下命令,安装pip和virtualenv:
$ sudo apt-get update
$ sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7
$ sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n
  • 创建一个新的virtualenv环境:
$ mkdir tensorflow
$ virtualenv --system-site-packages ~/tensorflow # for Python 2.7
$ virtualenv --system-site-packages -p python3 ~/tensorflow  # for Python 3.n
  • T1-2.png
  • 激活virtualenv环境:
$ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh
**执行完这条命令后,你的命令行会变成如下字样:
(tensorflow)$
  • 确保pip已经安装(版本>=8.1):
(tensorflow)$ easy_install -U pip
  • 执行下列命令,在virtualenv 环境中安装TensorFlow:
(tensorflow)$ pip install --upgrade tensorflow      # for Python 2.7
(tensorflow)$ pip3 install --upgrade tensorflow     # for Python 3.n
(tensorflow)$ pip install --upgrade tensorflow-gpu  # for Python 2.7 and GPU
(tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU
  • 用一个小的程序来验证安装:
$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
**如果出现下列警告,可以忽略它:(它们仅表示在当前的电脑中,有一种更好的编译tensorflow库的方法)
*T1-3.png
  • 如果系统输出如下,则证明TensorFlow安装成功,可以写TensorFlow程序了!
  • T1-4.png

第一个TensorFlow程序

  • 输入exit( )命令,退出python环境
  • 执行以下命令:
(tensorflow) cloud@ubuntu:~$ mkdir ai
(tensorflow) cloud@ubuntu:~$ cd ai
(tensorflow) cloud@ubuntu:~/ai$ vi hello.py
  • 在hello.py文件中编写如下内容:
import tensorflow as tf
#先定义计算
hello = tf.constant('Hello,TensorFlow!')
#然后启动session开始计算
sess = tf.Session()
print sess.run(hello)
  • 执行命令python hello.py,运行这个脚本
  • 同样会看到刚才这个结果:
  • T1-4.png


>>>