ubuntu下安装cuda10.1,cudnn

首先声明，此博客是默认你已经安装好了nvidia的驱动，如果你没安装好驱动请采参考这里的前几步安装驱动

即你已经可以输入

nvidia-smi

并输出，查看你是否安装好驱动

下面进入正题

1、安装cuda

从上图右上角可以看出，我的cuda版本为10.1,因此我在官网下载的10.1（任意一个10.1都行）对应的版本，点进去可以看到如下图。


然后按着官网的步骤安装，（我下载的是runfile文件，最好下载这个，因为好多人这样说。）输入

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run

找到你的这个文件的下载位置，例如我的位置在我自己创建的ruanjian文件夹下

这里多说几句，上面的输入中末尾语句cuda_10.1.243_418.87.00_linux.run，其中10.1.243是指你的cuda版本，418.87指的是cuda10.1要求的最低驱动版本，如果你的驱动版本低于这个，最后应该会出问题。

从第一幅图也可看出我的驱动版本为430.64,因此原则上只要我安装的cuda版本的驱动低于430就行。因此我也可以安装10.0、9.1等低版本的cuda。然后在该文件夹下打开终端，输入

sudo sh cuda_10.1.243_418.87.00_linux.run

这时会出现下图

选择continue

然后会出现，driver installer页面，注意这里不要选择第一个选项，第一项是让你更新驱动的，不用更新，因此在第一个选项上按enter键，可以看到第一个选项的括号里没有X，表明你没有选择这项。然后就没有太多注意的点了。

安装完之后，配置环境变量：

sudo gedit  ~/.bashrc

在文件下方加入：

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

注意，这里第三行命令中的cuda-10.1应该为你自己安装的版本名字。

查看安装是否成功

nvcc -V

证明安装成功。
如果不成功，提示：

Command 'nvcc' not found, but can be installed with:

sudo apt install nvidia-cuda-toolkit

在终端输入下面的，source下环境变量：

source ~/.bashrc

然后在输入nvcc -V就可以了

第二步、安装cudnn

在官网下载对应你cuda的版本,选择10.1的版本，点进去

选择　cuDNN Library for Linux,下载后解压，找到下载的文件夹解压，打开终端，输入

sudo cp cuda/lib64/* /usr/local/cuda-10.1/lib64/
sudo cp cuda/include/* /usr/local/cuda-10.1/include/

查看版本，输入

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

证明安装成功。
如果没有输出，按下面这样。

新版本查看cudnn是否成功，输入下面：

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

如果遇到什么问题，可以下方评论，我看到就回复