我已经搜索了很多地方,但我得到的都是如何安装它,而不是如何验证它是否安装。我可以验证我的NVIDIA驱动程序是否安装,CUDA是否安装,但我不知道如何验证CuDNN是否安装。非常感谢您的帮助,谢谢!
PS。 这是一个caffe实现。目前没有启用CuDNN,一切都可以正常工作。
我已经搜索了很多地方,但我得到的都是如何安装它,而不是如何验证它是否安装。我可以验证我的NVIDIA驱动程序是否安装,CUDA是否安装,但我不知道如何验证CuDNN是否安装。非常感谢您的帮助,谢谢!
PS。 这是一个caffe实现。目前没有启用CuDNN,一切都可以正常工作。
当前回答
我有cuDNN 8.0,上面的建议对我都不起作用。所需的信息在/usr/include/cudnn_version.h中,因此
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
成功了。
其他回答
torch.backends.cudnn.version()
应该能成功
如何检查python代码:
from tensorflow.python.platform import build_info as tf_build_info
print(tf_build_info.cudnn_version_number)
# 7 in v1.10.0
torch.backends.cudnn.m.is_available()
Ubuntu 20.04LTS操作系统:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR
返回预期结果
获取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