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

时间:2016-04-18 00:21:32

标签: python cuda pycharm tensorflow cudnn

我在安装了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安装的问题吗? (这是sheisty - 我刚刚运行rsync -av ./include/ /usr/local/cuda/includersync -av ./lib64/ /usr/local/cuda/lib64,然后我重新运行ldconfig /usr/local/cuda/lib64)...或者代码中的拼写错误?

或者它可能来自其他地方? (或者,更好的是,是否有人知道修复?)

2 个答案:

答案 0 :(得分: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

答案 1 :(得分:1)

您使用的是cudnn的第5版

请参阅Github issue #1787

补丁#1794在master中合并,因此最新的tensorflow RC(r0.8)has support for it