【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 的过程