首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何验证CuDNN安装?

如何验证CuDNN安装?
EN

Stack Overflow用户
提问于 2015-07-09 18:58:39
回答 13查看 564.9K关注 0票数 237

我搜索了很多地方,但我得到的只是如何安装它,而不是如何验证它是否已经安装。我可以验证我的NVIDIA驱动程序是否已安装,以及是否安装了CUDA,但我不知道如何验证CuDNN是否已安装。非常感谢您的帮助,谢谢!

PS。

这是为了一个caffe实现。目前,在没有启用CuDNN的情况下,一切都正常工作。

EN

回答 13

Stack Overflow用户

回答已采纳

发布于 2015-07-10 19:56:28

安装CuDNN只需要将文件放在CUDA目录中。如果您在安装caffe时正确地指定了路由和CuDNN选项,那么它将使用CuDNN进行编译。

您可以使用cmake进行检查。创建一个目录caffe/build并从那里运行cmake ..。如果配置正确,您将看到以下行:

代码语言:javascript
运行
复制
-- Found cuDNN (include: /usr/local/cuda-7.0/include, library: /usr/local/cuda-7.0/lib64/libcudnn.so)

-- NVIDIA CUDA:
--   Target GPU(s)     :   Auto
--   GPU arch(s)       :   sm_30
--   cuDNN             :   Yes

如果一切都是正确的,只需运行make命令从那里安装caffe。

票数 50
EN

Stack Overflow用户

发布于 2016-05-02 08:56:47

CuDNN的安装只是在复制一些文件。因此,要检查是否安装了CuDNN (以及有哪个版本),只需检查这些文件。

安装CuDNN

步骤1:注册一个nvidia开发人员帐户和下载cudnn在这里 (大约80 MB)。您可能需要nvcc --version来获得您的cuda版本。

第二步:检查您的cuda安装在哪里。对大多数人来说,这将是/usr/local/cuda/。您可以使用which nvcc检查它。

步骤3:复制文件:

代码语言:javascript
运行
复制
$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

校验版本

你可能需要调整路径。请参见安装的步骤2。

代码语言:javascript
运行
复制
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

编辑:在以后的版本中,这可能是如下(阿里斯的学分)

代码语言:javascript
运行
复制
$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

备注

当你遇到错误时

代码语言:javascript
运行
复制
F tensorflow/stream_executor/cuda/cuda_dnn.cc:427] could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM

使用TensorFlow,您可以考虑使用CuDNN v4而不是v5。

通过apthttps://askubuntu.com/a/767270/10425安装它的Ubuntu用户

票数 220
EN

Stack Overflow用户

发布于 2017-09-13 14:20:49

我的回答显示了如何检查安装的CuDNN版本,这通常也是您想要验证的。首先需要找到已安装的cudnn文件,然后解析该文件。要查找该文件,可以使用:

代码语言:javascript
运行
复制
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)

如果这不起作用,请参见下面的"Redhat分发版“。

一旦找到这个位置,就可以执行以下操作(用路径替换${CUDNN_H_PATH} ):

代码语言:javascript
运行
复制
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2

结果应该是这样的:

代码语言:javascript
运行
复制
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

这意味着版本是7.5.0。

Ubuntu18.04(通过sudo apt安装nvidia-cuda-工具箱)

这种安装方法安装在/usr/include和/usr/lib/ cuda /lib 64中,因此您需要查看的文件是/usr/include/cudnn.h。

代码语言:javascript
运行
复制
CUDNN_H_PATH=/usr/include/cudnn.h
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2

Debian和Ubuntu

从CuDNN v5开始(至少当您通过sudo dpkg -i <library_name>.deb包安装时),您可能需要使用以下内容:

代码语言:javascript
运行
复制
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

例如:

代码语言:javascript
运行
复制
$ cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2                                                         
#define CUDNN_MAJOR      6
#define CUDNN_MINOR      0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

指示已安装CuDNN版本6.0.21。

Redhat分布

在CentOS上,我发现了数据自动化系统的位置:

代码语言:javascript
运行
复制
$ whereis cuda
cuda: /usr/local/cuda

然后,我在cudnn.h文件上使用了从这个位置找到的过程:

代码语言:javascript
运行
复制
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
票数 154
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/31326015

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档