我搜索了很多地方,但我得到的只是如何安装它,而不是如何验证它是否已经安装。我可以验证我的NVIDIA驱动程序是否已安装,以及是否安装了CUDA,但我不知道如何验证CuDNN是否已安装。非常感谢您的帮助,谢谢!
PS。
这是为了一个caffe实现。目前,在没有启用CuDNN的情况下,一切都正常工作。
发布于 2015-07-10 19:56:28
安装CuDNN只需要将文件放在CUDA目录中。如果您在安装caffe时正确地指定了路由和CuDNN选项,那么它将使用CuDNN进行编译。
您可以使用cmake
进行检查。创建一个目录caffe/build
并从那里运行cmake ..
。如果配置正确,您将看到以下行:
-- 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。
发布于 2016-05-02 08:56:47
CuDNN的安装只是在复制一些文件。因此,要检查是否安装了CuDNN (以及有哪个版本),只需检查这些文件。
安装CuDNN
步骤1:注册一个nvidia开发人员帐户和下载cudnn在这里 (大约80 MB)。您可能需要nvcc --version
来获得您的cuda版本。
第二步:检查您的cuda安装在哪里。对大多数人来说,这将是/usr/local/cuda/
。您可以使用which nvcc
检查它。
步骤3:复制文件:
$ 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。
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
编辑:在以后的版本中,这可能是如下(阿里斯的学分)
$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
备注
当你遇到错误时
F tensorflow/stream_executor/cuda/cuda_dnn.cc:427] could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM
使用TensorFlow,您可以考虑使用CuDNN v4而不是v5。
通过apt
:https://askubuntu.com/a/767270/10425安装它的Ubuntu用户
发布于 2017-09-13 14:20:49
我的回答显示了如何检查安装的CuDNN版本,这通常也是您想要验证的。首先需要找到已安装的cudnn文件,然后解析该文件。要查找该文件,可以使用:
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
如果这不起作用,请参见下面的"Redhat分发版“。
一旦找到这个位置,就可以执行以下操作(用路径替换${CUDNN_H_PATH}
):
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
结果应该是这样的:
#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。
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
包安装时),您可能需要使用以下内容:
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
例如:
$ 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上,我发现了数据自动化系统的位置:
$ whereis cuda
cuda: /usr/local/cuda
然后,我在cudnn.h文件上使用了从这个位置找到的过程:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
https://stackoverflow.com/questions/31326015
复制相似问题