Ubuntu16.04安装cuda及cudnn方法

准备工作

安装显卡驱动

具体步骤可以参考我这篇文章Ubuntu 16.04安装nvidia显卡驱动

下载cuda

我下载的是cuda10.0，在官方网站上找到对应版本下载即可。

下载完成之后，得到cuda_10.0.130_410.48_linux.run文件。

下载cudnn

我下载的是cudnn7.4.2，在官方网站上找到相应版本下载cuDNN Librarary for Linux即可，这可能需要你注册账号或者登录。

下载完成后，得到cudnn-10.0-linux-x64-v7.4.2.24.tgz文件。

安装cuda

sudo chmod a+x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run

其中依次输入accept，n(不装显卡)，y(安装cuda Toolkit)，y(添加符号链接)，y或n都可，然后回车执行安装。
安装完成之后，添加环境变量：

sudo gedit ~/.bashrc

添加：

export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

然后执行source ~/.bashrc使更改立即生效。

测试是否安装成功，新打开一个终端，输入：

nvcc -V

若出现版本信息，说明安装成功。

安装cudnn

将下载得到的cudnn-10.0-linux-x64-v7.4.2.24.tgz文件解压得到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 /usr/local/cuda/lib64/libcudnn*

即可完成安装。
测试是否安装成功，在终端中输入：

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

若出现版本信息，说明安装成功。

至此，全部大功告成。