cuda - 如何验证 CuDNN 安装？

Question

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

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

Answer 1

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

安装 CuDNN

第 1 步:注册 nvidia 开发者帐户并 download cudnn here (约 80 MB)。您可能需要 nvcc --version 来获取您的 cuda 版本。

第 2 步:检查您的 cuda 安装位置。对于大多数人来说，它将是 /usr/local/cuda/。您可以使用 which nvcc 进行检查。

第 3 步:复制文件:

$ 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

注释

当您收到类似错误时

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

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

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