Ubuntu16.04 下安装Cuda9.1+Cudnn7 及升级Cuda10

官网文档地址：
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，则卸载掉。

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -rf /usr/local/cuda-8.0

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

升级到Cuda10.1

注意tensorflow1.13要用Cuda10.0
下载新版本Cuda10.1

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -rf /usr/local/cuda-8.0

driver不要安装。install结束后，修改环境变量

vim ~/.bashrc

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

在终端运行：source ~/.bashrc

查看是否安装成功：nvcc --version查看cuda输出版本是否正确。

升级 cudnn

1.删除旧版本的cudnn

（如果找到.cuda-uninstaller可以直接运行 ./cuda-uninstaller）

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn

下载新版本libcudnn7_7.5.0.51-1+cuda10.1_amd64.deb

sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.1_amd64.deb