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

本文是 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 的过程

posted @ 2023-08-08 09:01  SeanSiyang  阅读(543)  评论(0编辑  收藏  举报