Ubutun安装tensorflow-gpu（包括配置显卡驱动和cuda）

踩坑两天，写个教程

一：显卡驱动安装

1.查看自己的电脑GPU型号：

lspci |grep -i nvidia

2.去英伟达官网下载对应型号的驱动（下载run后缀的，deb后缀的不完整）

http://www.nvidia.cn/Download/index.aspx?lang=cn

3.ubuntu默认安装了第三方开源的驱动程序nouveau，安装nvidia显卡驱动首先需要禁用nouveau，不然会碰到冲突的问题，导致无法安装nvidia显卡驱动。

编辑文件blacklist.conf

sudo vim /etc/modprobe.d/blacklist.conf

在文件最后插入两行

blacklist nouveau
 
options nouveau modeset=0

更新系统

sudo update-initramfs -u

重启（必须），验证nouveau是否已经禁用

lsmod | grep nouveau

没有信息显示，说明nouveau已被禁用，接下来可以安装nvidia的显卡驱动。

4.在ubuntu下按ctrl+alt+f6进入命令行界面(进入命令行的方式可能有所不同）

登录后在命令行输入：

sudo service lightdm stop      //这个是关闭图形界面，不执行会出错。（执行该语句时提示no lightdm service loaded ,跳过即可）

卸掉原有驱动：

sudo apt-get remove nvidia-*  （若安装过其他版本或其他方式安装过驱动执行此项）

5.安装

进入驱动所在文件夹，给run文件赋予权限并安装

sudo chmod  a+x NVIDIA-Linux-x86_64-410.78.run
sudo ./NVIDIA-Linux-x86_64-410.78.run -no-x-check -no-nouveau-check -no-opengl-files //只有禁用opengl这样安装才不会出现循环登陆的问题

-no-x-check：安装驱动时关闭X服务

-no-nouveau-check：安装驱动时禁用nouveau

-no-opengl-files：只安装驱动文件，不安装OpenGL文件

 

安装过程中如果报错，查看是否以下依赖包没有安装：

出现找不到cc文件的问题，要执行下面的语句，否则安装完成，建立软连接也无法找到cc文件

sudo apt-get install gcc
sudo apt-get install g++
sudo apt-get install make

6.检查驱动是否安装成功：

nvidia-smi

7.重启或者进入图形界面

重启：sudo reboot 重启

进入图形界面：sudo service lightdm start 或者键盘CTRL+ALT+F1

 

二.安装CUDA

1.去官网点击打开链接下载CUDA，需要注册一个账号（上外网登陆下载比较快）

https://developer.nvidia.com/cuda-toolkit-archive

注意：下载runfile（local）文件

2. 下载完成以后，找到CUDA10.0所在位置（我的是放在home目录下），终端输入：.

sudo chmod +x cuda_10.0.130_410.48_linux.run

sudo ./cuda_10.0.130_410.48_linux.run

安装过程中会有几个选项需要确认：

注意：不要安装cuda自带的显卡驱动，其他的直接选y，路径输入默认值即可

3. 配置环境

安装完成以后，需要把cuda路径添加到当前用户的配置文件里：

sudo gedit ~/.bashrc

将安装路径添加到文件末尾：

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

 然后执行如下命令使路径生效：

. ~/.bashrc

4. 验证

终端输入：

        nvcc -V

可以看到cuda的版本信息：

接着尝试运行一下cuda中自带的例子：

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

可以看到输出成功：

Result = PASS

三.CUDNN的安装

1.官网下载cuda对应版本的cudnn

https://developer.nvidia.com/rdp/cudnn-archive

2.下载完成以后将其解压到Cuda的目录当中，依次执行如下命令：

tar -xzvf cudnn-10.0-linux-x64-v7.6.0.64.tgz
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*

就这么简单ok啦

四.tensorflow安装

1.首先你可以使用virtualenvwrapper或着anconda管理你的python虚拟环境，到了后期你会发现他们超级有用

（我们的重点不在这里，自己下去学习哦！）

下面pip就可以直接安装了

pip install tensorflow-gpu==1.14       (注意tensorflow的版本）

如速度很慢的话可以设置pip的安装镜像源

终点等你哦！