ubuntu16.04安装nvidia驱动、cuda、cudnn

先说一下最关心的版本匹配的问题

（cudnn下载的时候就会对应cuda版本，cuda安装包名称会有比如说384就是对应的最低的驱动版本，高的驱动版本兼容低的）

电脑配置：显卡GeForce GTX 1070Ti

最近常用的版本：nvidia驱动410、396、390，cuda9.0、8.0，cudnn 7.1.4  5.1.10

 

驱动下载用这个网址： https://www.geforce.cn/drivers

 选择自己电脑的显卡配置，开始搜索，下面出现若干版本，自己选择需要的版本后下载

cuda下载网址：https://developer.nvidia.com/cuda-80-ga2-download-archive

这是下8.0的网址、点上级目录可选择下载其他版本的，下载 .run文件

cudnn下载网址：https://developer.nvidia.com/rdp/cudnn-archive

需要登录nvidia账号，我的密码是“963852ang，”  

 

下载完了需要的版本后开始安装

安装nvidia驱动

• 禁用nouveau驱动

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

#在文本最后添加：（禁用nouveau第三方驱动，之后也不需要改回来）

blacklist nouveau

options nouveau modeset=0

sudo update-initramfs -u

重启，然后执行：

lsmod | grep nouveau

如果没有屏幕输出，说明禁用nouveau成功。

• 关闭图形化界面

sudo service lightdm stop

按Ctrl-Alt+F1会进入命令行界面，输入用户名和密码登录

小提示：在当前命令行输入：sudo service lightdm start ，可恢复到图形化界面。

• 开始安装驱动

给驱动run文件赋予执行权限：

sudo chmod +x NVIDIA-Linux-x86_64-384.59.run 

sudo ./NVIDIA-Linux-x86_64-384.59.run --no-opengl-files

安装过程的选项 默认就行，装完后

nvidia-smi  

#若列出GPU的信息列表，表示驱动安装成功

安装cuda

图形化界面可以关、也可以不关

• 安装

sudo sh cuda_8.0.61_375.26_linux.run --no-opengl-libs

如果不能执行就赋予一下权限，sudo chmod +x cuda....

安装过程中：

ctrl+c 跳过阅读、accept、n（不安装驱动）、y （安装CUDA Toolkit）、<Enter> （安装到默认目录）、y （创建安装目录的软链接）、n （不复制Samples）

• 配置环境变量

sudo gedit ~/.bashrc

在该文件最后加入以下两行并保存：（8.0为例）

export PATH=/usr/local/cuda-8.0/bin:$PATH

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

使该配置生效：

source ~/.bashrc

安装完成

验证

nvcc -V

也可用此查看cuda版本

安装cudnn

下载完成后解压，得到一个 cudn 文件夹，该文件夹下include 和 lib64 两个文件夹，

cd cuda/include 

sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件

cd cuda/lib64   （5.1.10为例）

sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库 
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件 
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5 #生成软衔接 
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

安装完成

验证和查看版本

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

 

卸载cuda 

cd /usr/local/cuda/bin
sudo ./uninstall_cuda_9.1.pl

不放心也可以在执行一下 下面的

sudo apt-get remove cuda

sudo apt-get autoclean

sudo apt-get remove cuda*


cd /usr/local/

sudo rm -r cuda-9.1