如何验证 CuDNN 安装？

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

PS.
这是用于咖啡实现的。目前，在没有启用 CuDNN 的情况下一切正常。

CuDNN的安装只是复制一些文件。因此，要检查 CuDNN 是否已安装（以及您拥有哪个版本），您只需检查这些文件。

安装CuDNN

第 1 步：注册一个 nvidia 开发者帐户并在这里下载 cudnn（约 80 MB）。你可能需要nvcc --version获取你的cuda版本。

步骤 2：检查您的 cuda 安装位置。对于大多数人来说，这将是/usr/local/cuda/。你可以检查它which 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

编辑：在以后的版本中，这可能是以下内容（归功于Aris)

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

Notes

当你收到类似错误时

F tensorflow/stream_executor/cuda/cuda_dnn.cc:427] could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM

对于 TensorFlow，您可以考虑使用 CuDNN v4 而不是 v5。

通过以下方式安装的 Ubuntu 用户apt: https://askubuntu.com/a/767270/10425