显卡驱动安装(Ubuntu16.04 + cuda9.0 + cudnn7.4)

显卡驱动安装

配置环境：Ubuntu16.04 + cuda9.0 + cudnn7.0

1. cuda安装

上nvidia官网下载对应版本的cuda，在 https://developer.nvidia.com/cuda-toolkit 中找到对应的cuda版本，一般默认的是当前最新的版本，所以这里需要注意下载对应的版本。这里吐槽一下，cuda下载的地址真的不好找，若在主页中则需要通过： Home->ComputeWorks->CUDA Toolkit找到下载链接。我这里下载的是：cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb

1.1 关闭与nvidia内核不兼容的nouveau

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

1.2 该文件打开后是空的，在文件中写入：

	 blacklist nouveau
 	 blacklist lbm-nouveau
 	 options nouveau modeset=0
 	 alias nouveau off
  	 alias lbm-nouveau off

1.3 然后依次执行一下命令

	echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
	sudo update-initramfs -u
	sudo reboot

1.4 安装cuda

cd到cuda.deb所在的位置（下载的cuda文件，比如我这里下载的是：cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb）

	sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
	sudo apt-key add /var/cuda-repo-9-0-local/7f2af80.pub
	sudo apt-get update
	sudo apt-get install cuda

1.5 环境变量配置
在~/.bashrc文件中写入环境变量

	export PATH=$PATH:/usr/local/cuda/bin
	export LD_LIBRARY_PATH=:/usr/local/cuda/lib64

添加完环境变量之后，执行source ~/.bashrc 使其生效

1.6 验证cuda是否安装成功

	cd /sur/local/cuda/samples/1_utilities/deviceQuery
	sudo make
	sudo ./deviceQuery

启动deviceQuery后，若能查到GPU等相关信息，则表明安装成功。若提示找不到cuda的so库，有可能是环境变量没有加载成功，可以先退出终端，然后重新进入终端并执行sudo ldconfig使其生效，最终重新执行1.6步。

2. cudnn配置

下载cudnn需要先在nvidia官网注册开发账号：https://developer.nvidia.com/cudnn， 在当前主页则搜索路径为：Home -> Deep Learning -> Deep Learning Software -> NVIDIA cuDNN，在列表中找到与cuda版本对应的cudnn即可，一般有多个cudnn版本都支持同一版本的cuda，看自己需求下载其中一个即可，我这里下载的是： cudnn-9.0-linux-x64-v7.4.1.5.tgz
cd 到cudnn-9.0-linux-x64-v7.4.1.5.tgz所在的目录，一次执行：

	tar -zxf cudnn-9.0-linux-x64-v7.4.1.5.tgz
	cd cuda
	sudo cp lib64/* /usr/local/cuda/lib64/
	sudo cp include/cudnn.h /usr/local/cuda/include/

解压tgz文件时，可能会报错，此时可以忽略，只需要能把lib64/*、cudnn.h 文件解压出来就可以。

我按上述的说明安装一遍搞定，也祝大家安装一切顺利哈。