526互联

【Ubuntu】Cuda10.2与cuDNN7.6.5的安装

发布时间 2023-08-08 09:09:58作者: SeanSiyang

本文是 Cuda10.2 与 cuDNN7.6.5 安装记录，系统环境是 Ubuntu18.04

所使用的显卡是 GeForce RTX 2080，因为不是30系的显卡，所以 Cuda 安装 10.2 就足够了

因为项目需要，要配置一下深度学习环境，一直没有整理和总结配置过程，就想记录一下，如果有错误的地方还请大家批评指正

Cuda10.2的安装

安装步骤：

  • 下载 runfile 文件
  • 运行 runfile 文件
  • 设置环境变量
  • 检查安装是否成功

下载runfile文件

首先，点击右侧链接下载 Cuda10.2：下载地址

点开链接后，按下图的方式选择安装方式：

依次选择：Linux-->x86_64-->Ubuntu-->18.04-->runfile(local)

选择以后，在下方弹出的界面里有两条命令，打开终端，执行这两行命令：

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

第一行命令是用于下载 Cuda 10.2

第二行命令是用于安装 Cuda 10.2，需要在文件所在目录打开终端后，再执行第二条命令

使用runfile安装Cuda

执行安装命令时，首先会弹出是否同意协议，此次输入 accept，并且回车：

Do you accept the above EULA? (accept/decline/quit):
accept

之后会进入安装内容选择界面，此处需要注意，不要安装驱动，将Driver选项去掉以后，选择Install:

在等待片刻后，就可以完成 Cuda10.2 的安装

环境变量的配置

安装好以后，需要配置环境变量，打开终端，执行下面的命令：

sudo gedit ~/.bashrc

在打开的界面中，在最后面添加下面两行内容：

export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64

添加好以后，保存退出gedit，并输入以下命令：

source ~/.bashrc

检查安装是否成功

此时可以选择重启，也可以直接测试是否安装成功：

nvcc -V

如果输出下面的内容，代表安装与环境配置都已经完成：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

Cudnn的安装

安装步骤：

  • 下载压缩包
  • 安装Cudnn（解压并复制文件）
  • 检查是否安装成功

下载压缩包

在完成Cuda的安装以后，就可以安装cuDNN了

如果使用过pytorch，会发现在安装Cudatoolkit时，如果选择10.2的版本，一般 cuDNN 配套都是7.6.5。因此，本文以安装 cuDNN 7.6.5 为例

首先进入 cuDNN 的网站，下载对应的版本：下载地址

需要检查下载的版本是否对应，下载以后文件名为：cudnn-10.2-linux-x64-v7.6.5.32.tgz

文件大小：548.2MB

这里需要注意，如果使用官网给的wget去安装，可能下载不完整，导致只有几十kb，那个版本是无法正常安装与使用的

安装cuDNN

下载以后，打开终端，进入文件所在目录，执行以下命令：

tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

之后目录下会生成一个 cuda 目录

执行以下命令，用于完成 cuDNN 的安装：

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

如果输出下面内容，代表安装成功：

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

综上，这就是 Ubuntu18.04 下，安装 Cuda10.2 和 cuDNN7.6.5 的过程