ubuntu16搭建深度学习环境(二)——安装cuda和cudnn

1、下载对应的CUDA

前提nvidia驱动已安装好,查看教程一:
查看cuda和驱动的对应(如图):官方cuda和驱动对应
在这里插入图片描述我安装的驱动是440.100,所以应选择CUDA10.2.89。到CUDA下载,拉到页面最底下选择download now,可以看到旧的cuda版本(如图)。

在这里插入图片描述

2、安装CUDA

quick start guide
选择CUDA Toolkit10.2,看到下载页面如下,在终端输入提示的命令即可。
在这里插入图片描述注意如果出现installation failed,则需要先修改run文件权限:

sudo chmod a+x cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

然后就开始进入安装,在一大堆文字后输入accept表示接受协议。
在这里插入图片描述提示incomplete installation这里不用管。
配置cuda环境变量:

sudo su //切换为root超级用户的命令
sudo gedit ~/.bashrc //打开.bashrc文件
//在~/.bashrc的最后添加:
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}}

添加之后一定要更新:source ~/.bashrc

3、测试CUDA是否安装成功

nvcc --version

成功后显示
在这里插入图片描述执行一个samples中的程序检测一下:

cd /home/hy/NVIDIA_CUDA-10.2_Samples/1_Utilities/deviceQuery  //输入自己的samples文件夹路径
sudo make
./deviceQuery

最后结果result=PASS应该就算成功了。
在这里插入图片描述

4、安装cudnn

首先到官网下载cuDNN https://developer.nvidia.com/cudnn ,需要注册为开发者才能下载。选择对应cuda版本的cudnn。我的是cuda10.2,故选择cudnn8,点击cudnn library for linux(x86),下载cudnn-xxxx.tgz文件。
在这里插入图片描述解压文件:

tar xvzf cudnn-10.2-linux-x64-v8.0.1.13.tgz

解压完成后得到一个cuda文件夹,该文件夹下有include和lib64两个文件夹。切换到cuda/include路径下,进行以下操作:

sudo cp cudnn.h /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/include/cudnn.h

切换到cuda/lib64:

sudo cp lib* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

5、检查cudnn版本

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

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 1

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include “driver_types.h”

猜你喜欢

转载自blog.csdn.net/qq_43265072/article/details/107161466
今日推荐