Linux Ubuntu18.04环境下cuda 10.0+NVIDIA驱动配置

前言

记录自己配置cuda的过程中遇到的坑，备忘。基本参照cuda官方文档。

1、准备工作（检查显卡型号，安装gcc）

lspci  | grep -i vga

我的电脑结果如下：

ltong@ltong-ThinkPad-T460:~$ lspci  | grep -i nvidia
06:00.0 3D controller: NVIDIA Corporation GM108M [GeForce 940MX] (rev a2)

型号为GF 940MX。可以到官网查看该型号的显卡是否支持CUDA，但一般NVIDIA的独显都会支持cuda。
需要安装gcc。

sudo apt install gcc

官网教程让我们更新内核

sudo apt-get install linux-headers-$(uname -r)

但是后面还是遇到了内核问题，通过安装dkm解决。

2、安装驱动

CUDA tookit中含有NVIDIA驱动，但是对于某些版本的显卡来说(比如我的显卡)，该驱动是不能用的。在你输入NVIDIA-smi的时候，会提示你检查是否正确安装了NVIDIA Driver。需要手动安装驱动。

根据你的显卡型号到NVIDIA官网去寻找正确的驱动（驱动下载链接）进行安装（这个过程比较容易出问题 ）。

1、禁用nouveau。
nouveau是Linux自带的显卡驱动，没有卵用并且会造成冲突，必须关闭 。
官网关闭nouveau教程如下：
建立文件 /etc/modprobe.d/blacklist-nouveau.conf，其内容如下
blacklist nouveau
options nouveau modeset=0
然后执行命令

$ sudo update-initramfs -u

该操作成功的标志为：

$ lsmod | grep nouveau

不产生任何输出。

2、关闭图形界面
NVIDIA驱动安装需要关闭图形界面，Ubuntu18.04操作如下

$ sudo systemctl