ubuntu 安装驱动及CUDA教程

ubuntu 安装驱动及CUDA教程
一.检查
若电脑已经有驱动，必须先删除原来驱动
使用以下指令
#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*

#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run –uninstall
若没有驱动，则无需这一步

二.查找
lspci | grep -i nvidia
查看显卡型号
sudo dpkg --list | grep nvidia-*
查看驱动版本

三.下载驱动以及CUDA
从nvidia官网下载对应版本驱动和CUDA 网址链接https://www.nvidia.com
本人下载的驱动是NVIDIA-Linux-x86_64-390.87.run（首先找到显卡型号并查看对应版本的驱动，如果版本错误后面执行驱动文件也会提示需要安装哪个版本正确），下载的CUDA是CUDA9.0，

四.禁用nouveau驱动
执行以下指令
sudo gedit /etc/modprobe.d/blacklist.conf

在文本最后添加：（禁用nouveau第三方驱动，之后也不需要改回来）
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

再执行
sudo update-initramfs -u

重启电脑

执行
lsmod | grep nouveau
如果无错误，说明禁用nouveau成功

五.禁用X-Window服务
sudo service lightdm stop
这会关闭图形界面,出现以下界面

按Ctrl-Alt+F1进入命令行界面，如果直接输入指令会出现login incorect
所以先输入电脑用户名，按enter键输入密码，即可以像终端一样输入指令

六.命令行安装驱动
输入以下指令
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
（给驱动run文件赋予执行权限）
sudo ./NVIDIA-Linux-x86_64-390.87.run –no-opengl-files
（后面的参数非常重要，不可省略）

然后一直按enter键，安装驱动成功后，输入
sudo service lightdm start，恢复到图形界面，即电脑主页面

在执行以下命令验证是否安装成功
nvidia-smi （若列出GPU的信息列表，表示驱动安装成功）

七.安装CUDA
执行下面命令
sudo ./ cuda_9.0.176_384.81_linux.run—no-opengl-libs

之后按照提示依次输入
accept （同意安装）
n （不安装Driver，因为已安装最新驱动）
y （安装CUDA Toolkit）
点击 （安装到默认目录）
y （创建安装目录的软链接）
n （不复制Samples，因为在安装目录下有sample）

安装后检验是否成功，输入
cat /usr/local/cuda/version.txt（列出CUDA版本即是安装成功）

八.安装cudnn
从invidia官网下载cudnn，

下载之后，解压cudnn压缩包文件
tar -xzvf cudnn-9.0-linux-x64-v7.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*

安装完成后，执行以下指令检验是否成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
出现cudnn版本显示，即成功安装

全文结束，感谢借鉴。