Ubuntu 16.04 + cuda-8.0 + cudnn-6.0 + Tensorflow(及其简单)

简介

因深度学习的需要，跑模型需要GPU加速，于是自己配置了一下，在实验室学长的帮助下，过程曲折，但其实很简单。get 几点经验:

1. 网上教程需要自己配置NVIDIA显卡驱动,其实不需要,找到”系统设置”->”软件与更新”->”附加驱动”->选择使用”NVIDIA binary driver”,点击更改。会自动下载好所需要的NVIDIA显卡驱动。
   
2. 切记下载的cuda文件别损坏，不然dpkg -i安装会失败。
3. ubuntu 16.04安装进入黑屏的话，按e进入配置，在”quiet splash”后面加上nomodeset, “quiet splash nomodeset”。
4. 最开始安装Ubuntu 17.10 + cuda-9.0出现问题，原因是pip3下载安装的Tensorflow最高支持到Cuda-8.0，估计过段时间就能OK。

版本介绍

• Ubuntu 16.04 LTS
• NVIDIA驱动 384.90
• Cuda-8.0
• Cudnn-6.0
• Tensorflow-gpu 1.4.0

Cuda-8.0 安装

1.官网链接，下载好了，安装截图命令安装就行。
cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb


2.安装完之后，配置环境变量，打开/etc/profile

$ sudo gedit /etc/profile

将下面两句添加到文件尾保存。

$ export PATH=/usr/local/cuda-8.0/bin:$PATH$ export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64

3.测试例子，成功后会显示配置信息。

$ cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery$ sudo make$ ./deviceQuery

Cudnn-6.0安装

官网链接，cudnn6.0下载好之后解压，会出现cuda文件夹。
选择”cuDNN v6.0 Library for Linux”

下面将cuda文件夹下面的东西拷贝到cuda即可，命令如下:

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include$ sudo cp cuda/lib64/* /usr/local/cuda/lib64/

Tensorflow安装

官网上就两步，官网指导链接如下，–upgrate第一次安装可不带,成功后截图。
Installing Tensorflow for Ubuntu

$ sudo apt-get install libcupti-dev$ pip3 install --upgrade tensorflow-gpu

结束语

网上教程一大堆，最简单的也就是这样了，最麻烦事下载文件花时间，之前在百度云分享上下载的cuda-8.0xx.deb文件损坏，导致了郁闷，官网下最靠谱。总之有什么不对的地方，请指正。