Ubuntu 14.04,CUDA版本7.5.18,每晚构建张量流
在tensorflow中运行 tf.nn.max_pool()
操作时,我收到以下错误:
E tensorflow / stream_executor / cuda / cuda_dnn.cc:286]加载的cudnn库:5005但源代码是针对4007编译的 . 如果使用二进制安装,请升级您的cudnn库以匹配 . 如果从源构建,请确保加载的库与您在编译配置期间指定的版本匹配 . W tensorflow / stream_executor / stream.cc:577]尝试使用StreamExecutor执行DNN操作而不支持DNN Traceback(最近一次调用最后一次):...
如何在tensorflow的编译配置中指定我的cudnn版本?
4 回答
进入TensorFlow源代码目录,然后执行配置文件:
/.configure
.以下是TensorFlow documentation的示例:
好像你已经安装了cudnn 5 . 你需要在运行
./configure
时设置它添加我的2美分:在我的情况下(TF0.12.1,从
pip
安装到anaconda,没有sudo
权限)CuDNNv5已安装,但不是默认值 .设置
export LD_LIBRARY_PATH="/usr/local/lib/cuda-8.0/lib64:/usr/local/lib/cudann5/lib64/"
解决了这个问题我也遇到了这样一个不兼容的问题:
所以我下载了CuDNN 5.1(与CUDA8.0兼容)并用它替换5.0然后一切顺利 .
警告:来自nvidia的CuDNN是不可用的,但您可以从其他人的共享中找到它 .