Ubuntu16.04 下安装Cuda9.1+Cudnn7

官网文档地址：

http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#ubuntu-installation

一、系统需求

1. 查显卡型号是否支持

lspci | grep -i nvidia

找到VGA compatible controller型号，并到Cuda网站查询。

https://developer.nvidia.com/cuda-gpus

2. Linux 版本验证

3. gcc 版本验证

4. Kernel Headers和开发包验证

目前Ubuntu16.04这些是支持的。

5. 选择安装方式

https://developer.nvidia.com/cuda-downloads

我使用的是离线下载。

6. 如果之前已经安装了Cuda，则卸载掉。

7. 禁用nouveau

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

内容：

blacklist nouveau
options nouveau modeset=0

更新系统：

sudo update-initramfs -u

重新启动，验证下面一句无输出则为成功：

lsmod | grep nouveau

禁用图形界面

sudo service lightdm stop

按ctrl+alt+f1进入命令行界面。

上面操作如果不行，则直接移除：

cd /lib/modules/4.10.0-28-generic/kernel/drivers/gpu/drm/nouveau
sudo rm -rf nouveau.ko
sudo rm -rf nouveau.ko.org

8. 安装nvidia驱动

下载驱动：

http://www.geforce.cn/drivers

卸载原驱动

sudo apt-get remove --purge nvidia-*

我在操作的时候，不幸遇到：

ubuntu 16 ImportError: No module named 'ConfigParser'

处理方法：把python先切回python2

rm /usr/bin/python
ln -s /usr/bin/python2 /usr/bin/python
# 删除python-configparser，再通过aptitude来安装
sudo apt-get purge --auto-remove python-configparser
sudo aptitude install python-configparser

再卸载原驱动。

安装新下载的驱动

sudo chmod a+x NVIDIA-Linux-x86_64-384.111.run //获取权限
sudo ./NVIDIA-Linux-x86_64-384.111.run -no-x-check -no-nouveau-check -no-opengl-files //安装驱动
sudo reboot
service lightdm restart

–no-x-check 安装驱动时关闭X服务

–no-nouveau-check 安装驱动时禁用nouveau

–no-opengl-files 只安装驱动文件，不安装OpenGL文件

验证：

nvidia-smi

安装cuda

sudo sh ./cuda_9.1.85_3

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 387.26?

要选否。

安装默认位置： /usr/local/cuda-9.1

samples位置： /home/admin1

下载安装Cudnn7.0.5 For Cuda9.1

https://developer.nvidia.com/rdp/cudnn-download#a-collapse704-9

测试cuda demo

cd /home/admin1
mkdir test
cd test
cp -r /home/admin1/NVIDIA_CUDA-9.1_Samples/ .
cd NVIDIA_CUDA-9.1_Samples
make