通过Anaconda在Ubuntu16.04上安装 TensorFlow(GPU版本)

一. 安装环境

• Ubuntu16.04.3 LST
• GPU: GeForce GTX1070
• Python: 3.5
• CUDA Toolkit 8.0 GA1 (Sept 2016)
• cuDNN v6.0 Library for Linux
• TensorFlow版本： Linux GPU:  Python 3.5 (build history)

版本之间要匹配，否则安装可能会出错。

 

二、软件下载：

1、Ubuntu16.04.3 LST

• 下载地址：https://www.ubuntu.com/download/desktop

2、CUDA Toolkit 8.0 GA1 (Sept 2016)

• 下载地址：https://developer.nvidia.com/cuda-toolkit-archive
• 下载界面截图：

 

3、cuDNN v6.0 Library for Linux

• 下载地址：https://developer.nvidia.com/rdp/cudnn-download  （下载前需要注册下，可以用新浪邮箱注册，qq邮箱不能识别）
• 下载界面截图：

 

4、TensorFlow版本： Linux GPU:  Python 3.5 (build history)

• 下载地址：https://github.com/tensorflow/tensorflow
• 下载界面截图：

 

三、软件安装

1、安装NAVID驱动

1）打开terminal输入以下指令:

sudo apt - get update

2）然后在系统设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改 。如下图所示：

3）在终端输入以下命令，查看安装的驱动版本：

cat / proc / driver / nvidia / version

版本信息如下图所示：

 

注释：也可以自己到官网下载适合自己显卡的驱动

• 下载地址：http://www.nvidia.com/Download/index.aspx?lang=en-us
• 下载界面截图：

 

2、安装CUDA Toolkit 8.0 GA1

在终端，进入到CUDA Toolkit 8.0 GA1下载的目录,然后执行以下命令安装：

sudo dpkg - i cuda - repo - ubuntu1604 - 8 - 0 - local_8. 0.44 - 1_amd64 .deb   sudo apt - get update   sudo apt - get install cuda

 

3、安装cuDNN v6.0 Library for Linux　　（详细的安装参考：http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html ）

进入到cuDNN v6.0 Library for Linux下载的目录，然后执行下列命令：

（1）解压 cuDNN安装包

tar - xzvf cudnn - 8.0 - linux - x64 - v6. 0.tgz

 （2）复制下列文件到CUDA Toolkit对应的目录下：

sudo cp cuda / include / cudnn.h / usr / local / cuda / include sudo cp cuda / lib64 / libcudnn *  / usr / local / cuda / lib64 sudo chmod a + r / usr / local / cuda / include / cudnn.h / usr / local / cuda / lib64 / libcudnn *

 

4、采取以下步骤在Anaconda环境中安装 TensorFlow:

（1）按照Anaconda下载网站上的说明下载并安装Anaconda。

（2）创建一个名为 tensorflow 的conda环境, 通过调用以下命令来运行 Python 版本:

conda create - n tensorflow python = 3.5 . 2      #或者其他版本

（3）通过发出以下命令来激活conda环境:

source activate tensorflow

（4）发出以下格式的命令, 以便在您的conda环境中安装 TensorFlow:

pip install - - ignore - installed - - upgrade tfBinaryURL

其中 tfBinaryURL 是 TensorFlow Python 包的 URL。

例如, 下面的命令为 Python 3.4 安装 TensorFlow 的 CPU 版本:

pip install - - ignore - installed - - upgrade https: / / storage.googleapis.com / tensorflow / linux / cpu / tensorflow - 1.4 . 1 - cp34 - cp34m - linux_x86_64.whl

下面的命令为 Python 3.5 安装 TensorFlow 的 GPU 版本:

1	pip install  - - ignore - installed  - - upgrade https: / / storage.googleapis.com / tensorflow / linux / gpu / tensorflow_gpu - 1.4 . 1 - cp35 - cp35m - linux_x86_64.whl

 

【遇到的问题】

在执行上述命令时，安装过程中遇到的问题：PermissionError: [Errno 13] 权限不够

通过命令ls -ld /home/heyun/.conda/  查看文件夹权限，发现用户没有写的权限，如下图所示：

然后修改文件夹/home/heyun/.conda/及其子文件夹和子文件的权限，使用命令 chmod -R 777 /home/heyun/.conda/ ，如下图：

修改后再次查看该文件夹权限，用户已经具有写权限：

重新执行以下命令，成功安装，如下图：

pip install - - ignore - installed - - upgrade https: / / storage.googleapis.com / tensorflow / linux / gpu / tensorflow_gpu - 1.4 . 1 - cp35 - cp35m - linux_x86_64.whl

 

 

（5）确保名叫tensorflow的环境已经被成功添加：conda info --envs  ，从下图可以看到，环境已经添加上了。

 

（7）如果退出当前的环境，可以用下面的命令：

source deactivate

 

（8）补充不同python版本的TensorFlow的URL

Python 2.7

• CPU only:

https: / / storage.googleapis.com / tensorflow / linux / cpu / tensorflow - 1.4 . 1 - cp27 - none - linux_x86_64.whl

• GPU support:

https: / / storage.googleapis.com / tensorflow / linux / gpu / tensorflow_gpu - 1.4 . 1 - cp27 - none - linux_x86_64.whl

 

Python 3.4

• CPU only:

https: / / storage.googleapis.com / tensorflow / linux / cpu / tensorflow - 1.4 . 1 - cp34 - cp34m - linux_x86_64.whl

• GPU support:

https: / / storage.googleapis.com / tensorflow / linux / gpu / tensorflow_gpu - 1.4 . 1 - cp34 - cp34m - linux_x86_64.whl

 

 

Python 3.5

• CPU only:

https: / / storage.googleapis.com / tensorflow / linux / cpu / tensorflow - 1.4 . 1 - cp35 - cp35m - linux_x86_64.whl

• GPU support:

https: / / storage.googleapis.com / tensorflow / linux / gpu / tensorflow_gpu - 1.4 . 1 - cp35 - cp35m - linux_x86_64.whl

 

Python 3.6

CPU only:

https: / / storage.googleapis.com / tensorflow / linux / cpu / tensorflow - 1.4 . 1 - cp36 - cp36m - linux_x86_64.whl

GPU support:

https: / / storage.googleapis.com / tensorflow / linux / gpu / tensorflow_gpu - 1.4 . 1 - cp36 - cp36m - linux_x86_64.whl

 

5、验证安装是否成功

要验证您的 TensorFlow 安装, 请执行以下操作:

• 确保您的环境准备好了，然后运行 TensorFlow 程序。
• 运行一个简短的 TensorFlow 程序。

（1）启动终端。

（2）激活Anaconda, 输入以下命令：

source activate tensorflow

 （3）运行一个简短的TensorFlow 程序

从 shell 中调用 python, 如下所示:

python

 在 python 交互 shell 中输入以下短程序:

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

 如果系统输出以下内容, 则TensorFlow 安装成功：

Hello, TensorFlow!

 

-------------------------------------------------------

参考链接：

1、在 Ubuntu 上安装 TensorFlow (官方文档的翻译)：http://www.cnblogs.com/hezhiyao/p/8331333.html

2、OSError:[Errno 13]Permission denied解决方法：http://blog.csdn.net/jiangjieqazwsx/article/details/47029477

3、ubuntu16.04下安装TensorFlow(GPU加速)----详细图文教程 ： http://blog.csdn.net/zhaoyu106/article/details/52793183