Ubuntu 18.04 安装 Cuda10.1 和 cuDNN 7.5.1

前言

之前的系统崩了，换了新的Ubuntu系统，版本号18.04，由于Pytorch 刚刚发布了新的1.1版本，并且支持CUDA 10 ，所以我在这里记录一下安装的过程。虽然是ubuntu18.04和cuda10 ，但安装过程是一样的，理论上这个教程适用所有版本。

需要的文件

1. CUDA 10 本地安装的deb 文件
   下载地址： 下载地址
2. cuDNN 7.5.1的deb 文件
   这里需要去NVIDIA的开发者官网注册一个账号，用账号登录之后才能继续下载。注册登录过程我们不表。登陆之后，可以点击下载地址，同意协议之后选择适配CUDA 10 的 cuDNN 7.5.1 下载，选择 cuDNN Runtime Library for Ubuntu18.04 (Deb) 和 cuDNN Developer Library for Ubuntu18.04 (Deb) 两个文件。

注意事项

这里我们不需要再去下载安装 Ubuntu 的Nvidia 驱动，因为在 CUDA10 的安装包文件中会有相应的驱动文件，之后安装的时候会看到，所以不需要安装驱动再安装CUDA。

CUDA安装步骤

1.在你的文件目录文件夹下打开终端，输入命令：

sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
#安装文件
sudo apt-key add /var//var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub
#安装公钥

现在解释一下以上两个命令：DPKG是Debian系系统的软件管理机制，dpkg是其命令。dpkg -i 是一个简单的安装命令。我们可以用 dpkg-deb -c 命令查看软件的目录，如下图所示：

 sudo dpkg-deb -c  cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb 

其中，我们看到了/var//var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub 这个文件。这个文件是原发发布的公钥文件， 实际安装时会读取deb文件的签名信息，和电脑的签名信息进行比对，签名相同就会进行安装，反正，会警告停止安装。所以我们要安装公钥。

2. 首先停止图形界面服务，命令如下

sudo systemctl stop gdm.service

此时电脑会黑屏，假如没有反应，再次执行以上命令。

停止图形界面的原因是我们需要安装nvidia的驱动，在之后的安装CUDA过程中，程序会自动根据安装依赖进行驱动安装，所以在此之前不需要单独安装驱动。

3.切换一个新的虚拟终端Ctrl+Alt+F[1～6]，登录后，执行如下命令：

sudo apt-get update
sudo apt-get install cuda

等待安装完成后重启系统即可。

cuDNN 安装

相较于CUDA，cuDNN的安装较为简单，我们可以直接双击两个程序包cuDNN Runtime Library for Ubuntu18.04 (Deb) 和 cuDNN Developer Library for Ubuntu18.04 (Deb)，也可以通过dkpg -i 安装。

设置环境变量

我们可以把CUDA的路径加入到环境变量中（实测不加在Pytorch下也是可以调用CUDA 的，但为了保险起见，推荐添加）。
把以下两行追加到~/.bashrc中。

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

安装后的测试

1.切换到测试目录下：

cd  /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery

2.进行宏编译

 sudo make

3.进行测试，运行文件

./deviceQuery

我们会看到如下结果，PASS表示安装成功通过。

deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 10.1, CUDA Runtime Version = 10.1, NumDevs = 1
Result = PASS