云网牛站
所在位置:首页 > Linux教程 > 在Deepin 20系统下安装CUDNN的方法

在Deepin 20系统下安装CUDNN的方法

2020-09-17 15:36:52作者:frank1126稿源:深度站

本文介绍在Deepin 20操作系统下安装CUDNN的方法。

 

1、需要安装驱动

Deepin 20正式版已经集成了NVIDIA闭源驱动,在安装操作系统的时候勾选安装闭源驱动即可。如果装Deepin 20系统的时候没有装,建议先搜索驱动,命令如下:

apt search xserver-xorg-video-nvidia

然后找到适合自己的驱动,使用apt命令安装:

sudo apt install nvidia-driver

然后安装nvidia-smi(如上一步测试已安装,请跳过):

sudo apt install nvidia-smi

另外推荐原来系统自带的驱动查看工具:

sudo apt install deepin-graphics-driver-manager

 

2、安装CUDA

最方便的方法就是在终端中运行以下命令:

sudo apt install nvidia-cuda-toolkit

可以安装到源内的CUDA。

但是目前这个版本是cuda 9.2(可以自己用sudo apt search nvidia-cuda-toolkit查询),要是想用10.2版,可参考在Deepin V20系统上安装CUDA 10.2的方法

在Deepin 20系统下安装CUDNN的方法

 

3. 安装CUDNN

从https://developer.nvidia.com/cudnn下载cuDNN Library for Linux,需要登陆。

下载解压之后,将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹:

sudo cp ./cuda/include/cudnn.h /usr/local/cuda/include

将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:

sudo cp ./cuda/lib64/* /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"

在Deepin 20系统下安装CUDNN的方法

至此,安装CUDNN完成。

 

结语

除了以上方法外,还可以用python,然后建议直接安装Anaconda建一个环境装cudnn,这样更加的方便。

 

相关主题

deepin15.7中安装tensorflow+cuda9.0+cudnn7的步骤

精选文章
热门文章