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的官方手册

欢迎大家交流学习~

以上

来自一只刚入坑的小白

发布了26 篇原创文章 · 获赞 17 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_39704651/article/details/79605585