Ubuntu14.04 安装 Tensorflow（转）

原文地址：http://www.rthpc.com/View?id=588

本文详细描述了 Ubuntu14.04 下安装 Tensorflow 的详细步骤。

 

    Tensorflow 使用 pip install 方式进行安装，需要由具有 sudo 权限的用户进行操作。

 

    安装 Tensorflow 的 GPU 版需要安装 CUDA 和 cuDNN，而且各 Tensorflow 版本对 CUDA 的版本是有要求的，下面列出我安装过的各 Tensorflow 及对应 CUDA、cuDNN 版本。

 

 

    详细操作步骤如下：

 

1. 正确安装Ubuntu14.04 系统；

 

2. 正确安装 CUDA（可参考文章手把手教你装 CUDA（Ubuntu14.04））；

 

3. 由具有 sudo 权限的用户 tester 进行安装工作，在 /home/tester 下新建 tensorflow 目录作为工作目录。

 

4. 使用 pip install 方式安装 Tensorflow，需要预装依赖软件。

 

    $ sudo apt-get install python-pip

    $ sudo apt-get install python-dev

 

5. 安装0.11.0 CPU 版；

 

    $ sudo pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

 

6. 运行 0.11.0 CPU 版；

 

    $ cd /usr/local/lib/python2.7/dist-packages/tensorflow/models/image/mnist

    $ time sudo python convolutional.py

 

7. 安装 0.11.0 GPU 版；

 

    $ sudo pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

 

8. 安装 cuDNN。将下载的 cudnn-8.0-linux-x64-v5.1.tgz 拷贝到工作目录下。

 

    $ tar –xvf cudnn-8.0-linux-x64-v5.1.tgz

    $ cd cuda

    $ cd include

    $ sudo cp cudnn.h /usr/local/include/.

        将 cudnn.h 拷贝到 /usr/local/include 目录下。

 

    $ cd ../lib64

    $ sudo cp libcudnn.so.5.1.10 /usr/local/lib/.

    $ sudo cp libcudnn_static.a /usr/local/lib/.

        将 2 个文件拷贝到 /usr/local/lib 目录下。

 

    $ cd /usr/local/lib

    $ sudo ln –s libcudnn.so.5.1.10 libcudnn.so.5

    $ sudo ln –s libcudnn.so.5 libcudnn.so

        进入 /usr/local/lib 目录做软链接。

    $ sudo ldconfig

 

9. 运行 0.11.0 GPU 版；

 

    $ cd /usr/local/lib/python2.7/dist-packages/tensorflow/models/image/mnist

    $ time sudo python convolutional.py

 

 需要说明的是，如果是首次安装，命令为：

 

    $ sudo pip install 安装文件名

 

如果以前安装过低版本，升级安装的命令为：

 

    $ sudo pip install --upgrade 安装文件名

 

 

 举例来说，前文所述是全新安装 0.11.0 版本的方法，如果以前安装过 0.8.0，那么此次升级安装0.11.0，安装命令为：

 

$ sudo pip install –upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

 

$ sudo pip install –upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

总结一下，Tensorflow 各版本运行时间如下：