ubuntu18.04.4LTS系统，安装cuda10.1与cudnn

一、安装cuda

1、下载安装包

首先去官网下载cuda安装包， 链接.

我找了半天cuda10.1,没找到，其实这样找：

1、进去之后，直接点击下载

2、点击下图所示：

3、下图中选择一个

4、选择deb(local)安装类型，然后下载。

2、安装

1、进入安装包所在的目录：

cd 安装包所在的目录

2、运行.deb文件，其实就是解压

sudo dpkg -i 你的.deb文件

第一次应该会让你安装一个类似：
sudo apt-key add /var/cuda-repo-/7fa2af80.pub这样的包，安装即可。然后再次运行上面的命令。

3、更新一下下载源：

sudo apt-get update

4、安装cuda

sudo apt-get install cuda

5、添加环境变量

sudo gedit ~/.bashrc

复制下面的内容到打开的文件的最后面：

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

这里注意，至于是不是cuda-10.1，去/usr/local/下看看，有的也可能不是。

最后使其生效：

source ~/.bashrc

6、重启

reboot

3、检测

命令行输入：

nvidia-smi

显示下面这样，说明驱动没问题。

输入下面命令，若显示下图所示，说明安装cuda成功。

nvcc -V

查看安装的cuda版本：

cat /usr/local/cuda/version.txt

二、安装cudnn

1、 下载cudnn压缩包

去cudnn的官网下载对应的版本：官网

普通机器 ：x86_64architecture 服务器机器 :Power architecture

选择 cuDNN Library for linux

按上图下载下来的是压缩包。要是选择其他的，应该是.deb文件，那就是另外的一种安装方式，我没试过，所以这里也就没写。

2、安装

将下载下来的压缩包解压（记得解压到home下，右键提取就是解压），当然也可以使用命令来解压：

sudo tar -xvzf cudnn-10.1-linux-x64-v7.6.5.32

解压到home下，可以看到cuda文件夹，在当前home目录打开终端，执行下面的命令：

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*

这就已经完成了安装。其实cudnn的安装，就是拷贝头文件和库文件到cuda 对应的文件夹下，并给予权限。

3、检测cudnn安装是否成功

查看cudnn的版本信息，在终端输入：

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