ubuntu16.04系统下CUDA8.0和CUDNN6.0和tensorflow(GPU)的安装

寒假一过，由于实验室电脑出了问题，进行了重装一切。

当然关键还在tensorflow的安装，要知道第一次摸索了一周才装好的啊。这次特地凭着一个月前的记忆并留心记录下了tensorflow的安装步骤，以供包括我你在内的大家学习~（虽然是第二次仍然有以前没有碰到的问题，难熬啊，安装成功一定要出去浪一波）

希望对亲们有帮助~

 

第一步 安装NVIDIA显卡驱动

1. 打开terminal（怎么打开？纳尼问度娘！），输入sudo apt-get update 更新系统（NVIDIA比较新的内核版本和ubuntu新版本有一定的对应关系；注意更新之后下次打开系统的时候在ubuntu选择页面选择新的内核的ubuntu，那个是新的？当然是数字大的那个就是啊）

2. 安装显卡驱动 打开 系统设置-软件更新-附加驱动，选择见图：

3.这里重启电脑！！！

打开NVIDIA X Server Settings,图像如下就是安装成功啦（网上其他验证方法都可以参考啦）

ps:如果中途哪里弄错了，还是可以卸载原驱动重新安装滴，卸载方法：

sudo apt-get remove --purge nvidia --*,重新安装请看第一步即可。

 

第二步 安装CUDA8.0

1.下载CUDA8.0

CUDA 8.0下载地址：

https://developer.nvidia.com/cuda-80-ga2-download-archive

选择linux的deb版本，选择如下图:

 

2. gcc降低版本

由于这里gcc5.0以上版本和将要安装的环境不匹配，而ubuntu16.04自带gcc5.0以上版本，所以这里可以先在终端输入:

gcc --version

从而查看gcc的版本，如果在5.0以下就请调到下一个小步骤。降版本请在终端输入下面的几行命令：

 

再在终端输入：

gcc --version

结果如下：

 

3. CUDA8.0的安装

在下载文件夹里打开终端。在终端输入如图所示的三行命令：

 

 

4. 配置环境变量：

先在终端执行如下命令打开配置环境变量的环境：

sudo gedit ~/.bashrc

再在打开的文档的最低端加上两行命令：

保存并关闭，执行如下命令使环境生效：

source ~/.bashrc

 

第三步 安装CUDNNv6.0

1. 下载CUDNNv6.0,网址链接如下：

https://developer.nvidia.com/rdp/cudnn-download#a-collapse6-8

这里需要账户登陆才可以下载，所以需要多一步注册。

CUDNN下载画面如下：

选择和我们要配置的匹配的，并打开小的菜单栏，如下图：

选中并下载上图中画下划线的文件。

 

2. 安装和验证

在下载目录下打开终端，执行下图四行代码进行解压其中一个压缩包并配置文件：

接下来解压剩下的三个安装包：

下面进行安装验证，如图：

显示test passed即说明CUDNN安装成功！！

 

第四步 安装tensorflow

1. 首先安装库(可以跳过)

sudo apt-get install libcupti-dev

 

2.pip 安装(根据自己要安装的python版本选择)

$ sudo apt-get install python-pip python-dev # for Python 2.7$ sudo apt-get install python3-pip python3-dev # for Python 3.n

 

3.tensorflow安装$ pip install tensorflow-gpu==1.4.0 # Python 2.7; GPU support$ pip3 install tensorflow-gpu # Python 3.n; GPU support

这里特别注意tensorflow的版本，根据官方文档（如下图），在博客里所说的配置下可以安装tensorflow1.4.0版本，博主就好几次因为安装了最新的tensorflow导致安装失败好几天找不到原因。

这里附上官方安装说明地址：

https://tensorflow.google.cn/install/install_linux

好啦，等待一会就安装好啦，接下来就是验证了。

 

第五步 安装验证

在终端输入 python ,即转入python输入，

然后输入代码:

[python] view plain copy

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

如果输出‘Hello, World!’即说明安装成功~

大功告成啦

 

参考：

https://tensorflow.google.cn/install/install_linux#ValidateYourInstallation

http://blog.csdn.net/fdqw_sph/article/details/78745375

http://blog.csdn.net/lyc_yongcai/article/details/73244388

http://blog.csdn.net/jiang_z_q/article/details/73264561

https://www.cnblogs.com/villow/p/tensorflow.html

http://blog.csdn.net/zhaoyu106/article/details/52793183

以及一些CUDA和CUDNN的官方手册

 

欢迎大家交流学习~

以上

来自一只刚入坑的小白