原文地址: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 各版本运行时间如下: