1.下载
注意:①下载cuDNN时,需要注册账号
②根据CUDA的版本来选择cuDNN版本
下载链接如下:https://developer.nvidia/rdp/cudnn-download
下载.tgz格式压缩包
2.安装
(1)进入文件所在位置,运行一下命令解压文件到当前目录
cd ~/downloads # 进入cudnn-10.0-linux-x64-v7.6.5.32.tgz所在位置
(2)分别复制一下文件cuda/include/cudnn.h与cuda/lib64/libcudnn*到CUDA Toolkit目录的include/下与lib64/下:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
(3)修改访问权限
sudo chmod 777 /usr/local/cuda/include/cudnn.h
sudo chmod 777 /usr/local/cuda/lib64/libcudnn*
(4)检查是否安装成功
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"
参考链接:
①. http://t.csdn/RM08y
②. http://t.csdn/8v2nQ
③. http://t.csdn/dymNH
④. http://t.csdn/7OyLw
更多推荐
linux安装cuDNN
发布评论