Ubuntu下CUDA和cuDNN的安装

介绍机场行李筐智能分拣项目的安装环境之CUDA和cuDNN的安装。

1 前言

在此之前,应当已经先装好了NVIDIA显卡驱动,并通过 nvidia-smi 指令查询到所装显卡对应的cuda版本。依据所安装的显卡版本或者显示的cuda version,从NVIDIA显卡驱动与CUDA版本对应网站查询到对应的CUDA版本。本文用的是CUDA 10.2以及cuDNN7.6.5。

2 CUDA 10.2的安装

以CUDA 10.2的安装为例,搜索引擎搜索cuda 10.2,进入官方下载页面,选择对应的系统版本等,出现如下页面。

01

2.1 下载cuda-10.2

下载cuda-10.2有两种方式:

  1. 按照官方给出的安装提示从终端执行:

    1
    wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
  2. 直接复制 wget 后的网址到浏览器地址栏,会弹出下载提示。

2.2 下载cuda-10.2补充包

从上图中直接点击Download (121.5MB)

2.3 安装CUDA-10.2

找到下载好的CUDA-10.2安装包,从其所在目录打开终端,执行

1
sudo sh cuda_10.2.89_440.33.01_linux.run

在第一个问题后输入 accept

02

按回车后,在出现的页面中通过向下移动光标,选中Driver,按回车取消安装这一行(这是默认的显卡驱动,之前已经安装过了,故不再安装),使其前面的[X]变为[ ]。再向下移动光标至最后一行,按回车开始安装。如下图所示。

03

04

2.4 安装CUDA-10.2补充包

从终端打开下载好的文件所在目录,执行

1
sudo sh cuda_10.2.1_linux.run

选择accept和默认安装即可。

05

06

07

2.5 环境变量的设置

执行

1
sudo gedit ~/.bashrc

在打开文件的最后添加以下内容:

1
2
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}             
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存文件后关闭,并执行

1
source ~/.bashrc

使得修改后的环境变量生效。

2.6 验证安装成功

执行

1
nvcc -V

出现cuda的安装版本,表示成功。

08

3 cuDNN7.6.5的安装

cuDNN是一个SDK,是一个专门用于神经网络的加速包,注意,它跟我们的CUDA没有一一对应的关系,即每一个版本的CUDA可能有好几个版本的cuDNN与之对应,但一般有一个最新版本的cuDNN版本与CUDA对应更好。

下载与CUDA版本相对应的cuDNN版本,将安装包解压之后执行

1
2
3
4
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*

执行

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

下图表示安装成功。

09

4 卸载

直接找到安装目录将其删除就好了。

1
2
3
cd /usr/local
sudo rm -rf cuda
sudo rm -rf cuda-10.2
------ 本文结束感谢您的阅读------
Donate a cup of cola?