Ubuntu利用virtualenv安装TensorFlow
来自CloudWiki
软硬件环境
- 操作系统:Ubuntu 16.04 LTS
- 硬件支持:TensorFlow with CPU support only.
这个文档是针对在没有NVIDIA GPU的机器上安装TensorFlow来写的。安装过程非常快(5-10分钟),所以即使你有一个NVIDIA GPU,也推荐先按这种方法安装。
- 在正式使用之前,如果你需要远程连接Ubuntu,则还需要在Ubuntu中安装ssh,具体请参考 Ubuntu不能远程连接的解决方案
安装方法
- 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
$ 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库的方法)
*![]()
第一个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)
>>>