如何验证CuDNN安装?

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

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

当前回答

对于CUDnn 8.1及以上版本，使用以下命令:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

其他回答

torch.backends.cudnn.m.is_available()

如何检查python代码:

from tensorflow.python.platform import build_info as tf_build_info

print(tf_build_info.cudnn_version_number)
# 7 in v1.10.0

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

torch.backends.cudnn.version()

应该能成功

我有cuDNN 8.0，上面的建议对我都不起作用。所需的信息在/usr/include/cudnn_version.h中，因此

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

成功了。

推荐文章