Linux下，Nvidia驱动以及Cuda的安装

最近刚好在看动手深度学习的书，需要在GPU环境下运行，故此记录一下。

一、确认安装环境

1.禁用nouveau(否则与NVIDIA驱动安装会有冲突，nouveau是linux自带系统驱动)

 右键打开终端
 sudo  vi /etc/modprobe.d/blacklist.conf（没有安装vim编辑器需要先执行 sudo apt-get install  vim）
 在打开的文件末尾添加以下两行代码
 blacklist nouveau
 options nouveau modeset=0
 保存退出
 update-initramfs –u   （重新生成 kernel initramfs）
 重启后（如果电脑有两个显卡驱动，图标不会变化；否则，图标会变大）打开终端输入一下命令：
 lsmod | grep nouveau （没有输出表示已经禁用）  

2.Nvidia驱动安装

A.执行如下命令将Ubuntu官方维护的NVIDIA显卡驱动软件源添加到系统中：

sudo add-apt-repository ppa:graphics-drivers/ppa 
sudo apt-get update

B.执行如下命令，查看适合自己电脑的显卡驱动，找到recommended对应项

sudo ubuntu-drivers devices

C找到电脑的设置>>软件和更新>>附加驱动>>选择对应的Nvidia驱动>>点击右下方按钮更改，完成后重启执行如下代码

nvidia-smi  测试驱动是否安装成功，成功则显示如下图：

二、Cuda安装

1.下载与驱动对应的cuda的.run文件

注意：下载好的.run文件移动到home目录下

2.安装

alt + ctrl + F1
sudo service lightdm stop     （表示在文字模式的安装cuda）
sudo  sh    cuda_8.0.61_375.26_linux.run/cuda_10.1.168_418.67_linux.run （自己下载好的文件名）

A 安装过程选项选择
由于自己在台式机和笔记本上分别安装cuda8.0和cuda10.1。
（1）对于cuda8.0会有一个很长的阅读声明，一直按住enter键直100%；而对于cuda10.1版本直接就是个图形化界面。阅读完成后，输入accept即可。
（2）cuda8.0中，选项install nvidia accelerated Graphics Driver,输入n，因为我们已经安装了nvidia的驱动；后面的选型均输入y，等待安装即可；
对于cuda10.1在它对应图形化界面取消该项，默认是【x】，ENTER键取消，选择install安装即可（为图形界面第一个选型）
说明：无论安装什么版本，我们都不安装Graphics Driver这一项即可（填n），其余选项均为 y
（3）如何最后看到有三行中，最后两行有两个installed，表示安装成功

B 输入reboot，重启电脑
C 相关配置

sudo vim ~/.bashrc  打开改文件

在文件最后添加以下两行：
export PATH=/usr/local/cuda-(对应版本号，如：10.1)/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-（同上）/lib64:$LD_LIBRARY_PATH12

保存后退到终端，执行如下命令：
source .bashrc

nvcc -V  检查是否安装成功

3测试

执行如下代码：
cd /usr/local/cuda/samples 
cd 1_Utilities/deviceQuery

sudo make
/deviceQuery

当看到Result=pass，则表示表示测试成功，如下图所示

参考相关博客：
https://www.linuxdashen.com/switch-intel-nvidia-graphics-card-ubuntu
https://www.linuxidc.com/Linux/2018-09/154292.htm