CudNN 错误 运行 TensorFlow:无法设置 cudnn 过滤器描述符:CUDNN_STATUS_BAD_PARAM

CudNN error running TensorFlow: Could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM

我在 Ubuntu 系统上使用 Python 和 IDE Pycharm,安装并运行 CUDA 7.5。

我刚刚将 CUDNN 文件与我的常规 CUDA 安装合并。

现在,当我 运行 来自 TensorFlow 网站的深度卷积 nn 教程时,我收到以下错误:

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

有谁知道这个问题的根源是什么?这是我的 CUDNN 安装的问题吗? (这是 sheisty——我只是 运行 rsync -av ./include/ /usr/local/cuda/includersync -av ./lib64/ /usr/local/cuda/lib64,然后我重新 运行 ldconfig /usr/local/cuda/lib64)...或者代码中的错字?


或者它可能来自其他地方? (或者,更好的是,有人知道解决方法吗?)

好的,我 运行 在互联网上的答案为零,所以我将保留它以防其他人遇到同样的问题。


问题是 cudnn v5 不能与张量流一起工作(也许一般情况下,也许只是在我的设置上......无可否认,它是 Ubuntu 15.10 而不是 CUDA 支持的 15.04),所以我不得不退回到 cudnn v4.

假设您在默认目录中安装了 CUDA 7.5 并在 /usr/local/cuda 中设置了符号 link,只需使用 CUDNN v4(从 2 月开始)即可使其正常工作。我刚刚提取了 CUDNN tar 文件和 运行 以下命令,具有 root 权限(来自我的下载文件夹):

tar -xvzf cudnn-etc.tgz
cd cuda 

rsync -av ./lib64/ /usr/local/cuda-7.5/lib64/
rsync -av ./include/ /usr/local/cuda-7.5/lib64/

ldconfig /usr/local/cuda/lib64

您使用的是 cudnn 版本 5

Github issue #1787

补丁#1794 is merged in master so the latest RC of tensorflow (r0.8) has support for it