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

PS。 这是一个caffe实现。目前没有启用CuDNN,一切都可以正常工作。


当前回答

在“/usr/src/cudnn_samples_v7/mnistCUDNN”目录下运行。/mnistCUDNN 这里有一个例子: cudnnGetVersion(): 7005, CUDNN_VERSION from cudnn.h: 7005 (7.0.5) 主机编译器版本:GCC 5.4.0 在您的机器上有1个CUDA功能设备: device 0: sms 30 Capabilities 6.1, SmClock 1645.0 Mhz, MemSize (Mb) 24446, MemClock 4513.0 Mhz, Ecc=0, boardGroupID=0 使用设备0

其他回答

当通过。deb在ubuntu上安装时,你可以使用sudo apt search cudnn | grep installed

要检查CUDA的安装,运行下面的命令,如果安装正确,下面的命令将不会抛出任何错误,并将打印正确的库版本。

function lib_installed() { /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep $1; }
function check() { lib_installed $1 && echo "$1 is installed" || echo "ERROR: $1 is NOT installed"; }
check libcuda
check libcudart

要检查CuDNN的安装,运行下面的命令,如果CuDNN安装正确,那么您将不会得到任何错误。

function lib_installed() { /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep $1; }
function check() { lib_installed $1 && echo "$1 is installed" || echo "ERROR: $1 is NOT installed"; }
check libcudnn 

OR

您可以从任何目录运行以下命令

nvcc -V

它的输出应该是这样的

 nvcc: NVIDIA (R) Cuda compiler driver
 Copyright (c) 2005-2016 NVIDIA Corporation
 Built on Tue_Jan_10_13:22:03_CST_2017
 Cuda compilation tools, release 8.0, V8.0.61

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

安装 CuDNN

第一步:注册一个nvidia开发者账号并下载cudnn(大约80 MB)。你可能需要nvcc -version来获得你的cuda版本。

步骤2:检查cuda的安装位置。对于大多数人来说,它将是/usr/local/cuda/。你可以用哪个nvcc检查一下。

第三步:复制文件:

$ 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用户

获取cuDNN版本[Linux]

使用以下方法查找cuDNN路径:

cat $(whereis cudnn.h) | grep CUDNN_MAJOR -A 2

如果以上都不行,试试这个:

cat $(whereis cuda)/include/cudnn.h | grep CUDNN_MAJOR -A 2

获取cuDNN版本[Windows]

使用以下方法查找cuDNN路径:

C:\>where cudnn*
C:\Program Files\cuDNN6\cuda\bin\cudnn64_6.dll

然后使用这个来从头文件中转储版本,

type "%PROGRAMFILES%\cuDNN6\cuda\include\cudnn.h" | findstr "CUDNN_MAJOR CUDNN_MINOR CUDNN_PATCHLEVEL"

获取CUDA版本

这在Linux和Windows上都有效:

nvcc --version

在“/usr/src/cudnn_samples_v7/mnistCUDNN”目录下运行。/mnistCUDNN 这里有一个例子: cudnnGetVersion(): 7005, CUDNN_VERSION from cudnn.h: 7005 (7.0.5) 主机编译器版本:GCC 5.4.0 在您的机器上有1个CUDA功能设备: device 0: sms 30 Capabilities 6.1, SmClock 1645.0 Mhz, MemSize (Mb) 24446, MemClock 4513.0 Mhz, Ecc=0, boardGroupID=0 使用设备0