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

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


当前回答

Ubuntu 20.04LTS操作系统:

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

返回预期结果

其他回答

如何检查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.version()

应该能成功

安装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。

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

Ubuntu 20.04LTS操作系统:

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

返回预期结果