搭建 TensorFlow 开发环境之 Hello World

纵然今后可能永远不会从事人工智能,但是这怎么能妨碍我学习它呢?毕竟,今后也可以在外面吹牛逼了:哥当年可是做(过) AI 的~

资源

安装的若干选择

初次接触 TensorFlow 的时候,我们可能会被文档里所提及的各种选择绕晕,为了避免被绕晕,我不多想,果断地做了如下的选择。

  1. 开发平台的选择。

    TensorFlow 支持 Ubuntu、Windows、和 MacOS 三大操作系统平台,此外,还支持通过源码进行安装。为了安装、开发的方便,我选择的是直接在 Ubuntu 上面使用官方提供的软件包进行安装。

  2. Python 的选择。

    TensorFlow 支持 Python2.x 和 Python 3.x 两个版本的 Python,我对这两个版本都不太熟,所以随便挑了一个 Python3.x。

  3. 是否选择使用 GPU。

    我的电脑本来就不支持 GPU,所以不用选择了,哈哈。不过,即便你的电脑支持 GPU,如果你是初学者对话,也建议不使用 GPU,相反可以等到你对 TensorFlow 比较熟悉后再使用 GPU 功能。

  4. 安装方式的选择。

    TensorFlow 提供了四种安装方式,分别是 virtualenv、native pip、Docker 以及 Anaconda,不过既然官方都推荐使用 virtualenv,我们还有什么好犹豫的呢。

具体的安装步骤

  1. 下面一条命令即可搞定 pip 及 virtualenv 的安装:

    1
    $ sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n
  2. 创建一个 virtualenv 环境。我是安装在 ~/tensorflow 目录下的,执行的是如下命令:

    1
    $ virtualenv --system-site-packages -p python3 tensorflow

    如果今后想删除这个 virtualenv 环境的话,直接删除这个安装目录(~/tensorflow)即可。

  3. 激活 virtualenv 环境。我习惯使用 bash shell:

    1
    $ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh

    激活成功后,控制台提示符会变成:

    1
    (tensorflow)$
  4. 在激活后的 eirtualenv 中安装 TensorFlow,输入如下的命令:

    1
    (tensorflow)$ pip3 install --upgrade tensorflow     # for Python 3.n

    这得稍微等一会儿,然后安装就完成了。

TensorFlow 之 Hello World

作为标准程序员,写过的 Hello World 不知有多少了,学 TensorFlow 怎么能少了 Hello World 呢。

需要注意的是,每次新打开一个控制台,都需要激活 virtualenv 环境:

1
$ source ~/tensorflow/bin/activate      # bash, sh, ksh, or zsh

激活后,我们再进行后续操作。

启动 Python 环境:

1
(tensorflow)$ python

在 Python 交互控制台中,依次输入如下代码:

1
2
3
4
5
# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

在输入完最后一条代码的时候,控制台会打印一条语句:

1
Hello, TensorFlow!

从此,又多了一个 Hello World 的光辉里程!