2

我在安装了 CUDA 7.5 并正常工作的 Ubuntu 系统上使用 Python 和 IDE Pycharm。

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

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

F tensorflow/stream_executor/cuda/cuda_dnn.cc:383] 无法设置 cudnn 过滤器描述符:CUDNN_STATUS_BAD_PARAM

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


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

4

2 回答 2

5

好的,我在互联网上遇到了零答案,所以我会留下这个,以防其他人遇到同样的问题。


问题是 cudnn v5 不适用于张量流(也许一般来说,也许只是在我的设置上......不可否认,它是 Ubuntu 15.10 而不是 CUDA 支持的 15.04),所以我不得不退回到 cudnn v4 .

假设您在默认目录中安装了 CUDA 7.5 并在 /usr/local/cuda 中设置了符号链接,只需使用 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
于 2016-04-18T02:27:35.530 回答
1

您使用的是 cudnn 版本 5

请参阅Github 问题 #1787

补丁#1794已合并到 master 中,因此 tensorflow (r0.8) 的最新 RC支持它

于 2016-04-18T20:24:43.260 回答