CUDA配置一篇就够——Ubuntu篇

前言

最近给Ubuntu服务器和CentOS配置升级环境，发现以前写的文章还是能使用，这里分享给大家。
Linux系统是相似的，配置centos也可以参照这篇。
AI环境主要指的是 cuda+cudnn等基础环境，这是搭建tensorflow和pytorch等神经网络框架的基础，caffe的配置则要相对麻烦很多，可以参考我别的文章。cuda和cudnn用对应的版本都行，都是相近的。

软件

• NVIDIA驱动安装地址
• CUDA安装地址
• cudnn下载——需要注册

系统安装

Ubuntu的安装在网上已经有很多教程，这里就不多做赘述。
参考系统安装
分盘的话，假如有128G硬盘，4G内存，个人倾向于

• / : 64G ， 相当于Window系统的C盘，各种软件安装在这里
• swap：4G， 直接等于内存大小即可
• home：剩下的都给home
• boot并不必要，真要分的话直接分2G即可

配置内核

通过官网的CUDA Toolkit可知，安装Ubuntu16.04需要4.4.0内核。
注意：当内核不匹配时，安装显卡驱动会出现错误。

当前查看内核版本

uname -r

如果你通过上面的命令显示出来的Ubuntu16.04内核如下为4.4.*版本，那么恭喜你，你可以跳过配置内核到下一步。

备份源配置文件（防止后面修改导致出错）

sudo cp /etc/apt/sources.list /etc/apt/sources.list_bak

编辑源配置文件

sudo apt install vim
sudo vim /etc/apt/sources.list

快捷键 shift + g 跳转到文件最后一行，按