Web安全之机器学习入门
上QQ阅读APP看书，第一时间看更新
上一章目录下一章

2.2 TensorFlow简介与环境搭建

TensorFlow是一个采用数据流图、用于数值计算的开源软件库（见图2-5）。节点在图中表示数学操作，图中的线则表示在节点间相互联系的多维数据数组，即张量。它灵活的架构使你可以在多种平台上展开计算，例如台式计算机中的一个或多个CPU（或GPU）、服务器、移动设备等等。TensorFlow最初由Google大脑小组（隶属于Google机器智能研究机构）的研究员和工程师们开发出来，用于机器学习和深度神经网络方面的研究，但这个系统的通用性使其也可广泛用于其他计算领域。

图2-5 TensorFlow中文社区主页

TensorFlow的特点：

● 高度的灵活性；

● 真正的可移植性；

● 将科研和产品联系在一起；

● 自动求微分；

● 多语言支持；

● 性能最优化。

安装方法如下：

    # 仅使用CPU的版本
    $ pip install https∶
    //storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-linux_
x86_64.whl
    # 开启GPU支持的版本 (安装该版本的前提是已经安装了CUDA sdk)
    $ pip install https∶
    //storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_
x86_64.whl

我们推荐使用virtualenv创建一个隔离的容器来安装TensorFlow，这样做能使排查安装问题变得更容易。

首先，安装所有必备工具：

    # 在Linux上∶
    $ sudo apt-get install python-pip python-dev python-virtualenv
    # 在Mac上∶
    $ sudo easy_install pip  # 如果还没有安装pip
    $ sudo pip install --upgrade virtualenv

接下来，建立一个全新的virtualenv环境。为了将环境建在～/tensorfl ow目录下，执行：

    $ virtualenv --system-site-packages ～/tensorflow
    $ cd ～/tensorflow

然后，激活virtualenv：

    $ source bin/activate  # 如果使用bash
    $ source bin/activate.csh  # 如果使用csh
    (tensorflow)$  # 终端提示符应该发生变化

在virtualenv内，安装TensorFlow∶

    (tensorflow)$ pip install -U TensorFlow

当使用完TensorFlow：

    (tensorflow)$ deactivate  # 停用virtualenv
上一章目录下一章