【问题标题】:CudNN error running TensorFlow: Could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM运行 TensorFlow 的 CudNN 错误:无法设置 cudnn 过滤器描述符:CUDNN_STATUS_BAD_PARAM
【发布时间】:2016-08-09 13:59:31
【问题描述】:

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

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

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

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)...或者代码中的错字?


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

【问题讨论】:

    标签: python cuda pycharm tensorflow cudnn


    【解决方案1】:

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


    问题是 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
    

    【讨论】:

    • 我在 16.04 LTS 上遇到了同样的问题。请注意,如果您从源代码编译 TF,您现在可以使用 cudnn v5。然而,TF 的 pip 版本仍然(截至 16 年 7 月)停留在 cudnn v4。似乎使用 Pascal GPU 的人应该从源代码安装,因为 Pascal 仅支持 cudnn v5 或更高版本。
    【解决方案2】:

    您使用的是 cudnn 版本 5

    Github issue #1787

    补丁#1794被合并到master中,所以tensorflow的最新RC(r0.8)has support for it

    【讨论】:

    • 好的,太好了,谢谢。完成项目后我会更新(现在一切正常,不会碰它:)
    猜你喜欢
    • 1970-01-01
    • 2022-06-18
    • 2019-02-01
    • 1970-01-01
    • 2016-10-06
    • 2019-04-02
    • 1970-01-01
    • 2019-09-24
    • 2022-12-12
    相关资源
    最近更新 更多