Ubuntu中CUDA和cuDNN的安装

CUDA的安装

1.版本的选择

TensorFlow1.0至1.4版本只支持CUDA8.0
TensorFlow1.5至1.12版本支持CUDA9.0
TensorFlow1.13之后的版本，支持10.0
更多版本CUDA：
https://developer.nvidia.com/cuda-toolkit-archive
TensorFlow1.0到1.2版本使用的是cuDNN5.1版本
TensorFlow1.3到1.4版本使用的是cuDNN6.0版本
TensorFlow1.5到1.10版本使用的是cuDNN7.0版本
TensorFlow1.11到1.12版本使用的是cuDNN7.2版本
TensorFlow1.13之后使用的是cuDNN7.5版本

2.禁用Nouveau驱动

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

在最后两行添加：

blacklist nouveau
options nouveau modeset=0

禁用nouveau第三方驱动，之后也不需要改回来

sudo update-initramfs -u #刷新内核

3.安装

或直接双击安装CUDA

cuDNN的安装

https://developer.nvidia.com/rdp/cudnn-archive

下载完成之后，解压cuDNN，将解压后的文件夹拷贝到CUDA安装的根目录下替换同名文件即可

或者输入下面代码（解压下载的文件，可以看到cuda文件夹，在当前目录打开终端，执行如下命令）

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*