Ubuntu16.04 配置显卡驱动+CUDA10.1+cudnn

基本说明

本篇主要基于ubuntu16.04进行gpu环境配置，gpu型号 1050Ti，请各位根据自己的显卡型号各取所需。

显卡驱动安装

1. 首先确认机器是否带有显卡

lspci | grep -i vga

带有N卡可进行下一步的nvidia 驱动安装 这里显示的1e82 设备对应Geforce1050Ti

2. 从官网下载.run进行安装
   对于未利用系统扩展方式安装驱动的情况，可以先到官方网站查看最新支持到的驱动版本
   nvidia中文官网驱动下载
   
   选择计算机对应的显卡

进行下载，假设保存于Downloads内

3. 禁用nouveau

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

在文件末尾添加

blacklist nouveau

4. 进入命令行界面，准备安装
   进行这个操作前，推荐各位记录好之前所做操作，以及准备好一个设备查看后续部分
   先ctrl+alt+F1进入非图形化界面，关闭图像化界面

sudo service lightdm stop

赋予可执行权限

cd Downloads
sudo chmod a+r NVIDIA-Linux-x86_64-430.50

如果没给上可执行权限，可以chmod + x试试

5. 进行安装
   为避免出现桌面循环登录，-no-x-check -no-nouveau-check -no-opengl-files 一定要有禁用 opengl部分

sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files

进安装状态，有几个选项
[1] Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 选择 No
[2] Nvidia’s 32-bit compatibility libraries? 选择 No
[3] Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择 No
安装完毕后，可开启并进入图形界面

sudo service lightdm start

终端内输入

nvidia-smi

可查看到刚刚安装的驱动

至此驱动安装完毕

另一种安装方法（不推荐）
依赖ubuntu系统自身扩展软件安装

关于这一类安装方法的驱动查看

sudo dpkg --list | grep nvidia-*

实际上是走的apt-get install 安装的，依赖于linux的源
当你想要卸载时

sudo apt-get remove --purge nvidia*

即可卸载由此方法安装的nvidia驱动

CUDA 10.1安装

首先先确认一件事情
在官网说明文当中
官网cuda toolkit说明文档
不同驱动版最高可按装的cuda版本不同，会导致最终环境有差别
下面开始cuda驱动安装
cuda官网下载驱动链接
选择好自己机器对应的型号
最后install type选择这里可以根据个人网速状态选择前两个，本文安装时选择了runfile

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run 

下载好后，依照官网的安装说明进行安装

sudo sh cuda_10.1.243_418.87.00_linux.run 

安装时，先出现是否同意协议的确认，输入accept
由于前面单独安装了最新的显卡驱动，这里

这样子选择即可，X代表要安装的部分
安装完成后，配置~/.bashrc

nano ~/.bashrc

在末尾添加

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc

即可查看刚刚完成安装的cuda版本

nvcc -V

cudnn下载

这里基本上只有下载和拷贝的操作，所以与其说是安装，不如说时环境配置的库添加
首先，在nvidia官网找到cudnn的对应版本，下载linux library *.tgz包
nvidia官网提供的 cudnn 地址
下载好了后，解压

tar -zxvf cudnn-9.0-linux-x64-v7.6.1.34.tgz

复制到cuda目录下

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*        #加权限

检查

 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

这里 cudnn7.6.4的就按装完毕了

参考链接