2016-10-04 76 views
7

我有以下错误。我正在使用tensorflow的conda安装。我努力尝试将它与我的GPU一起使用。加载运行时CuDNN库:5005(兼容版本5000),但源代码与5103(兼容版本5100)编译

Loaded runtime CuDNN library: 5005 (compatibility version 5000) but source was compiled with 5103 (compatibility version 5100). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration. F tensorflow/core/kernels/conv_ops.cc:526] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms) Aborted (core dumped)

这NVCC回报 /usr/local/cuda-7.5/bin/nvcc

NVCC版本返回 Cuda compilation tools, release 7.5, V7.5.17

我尝试下载CuDNN V5.1,做了以下的,但它没有工作,要么 ``` sudo cp lib * /usr/local/cuda-7.5/lib64/ sudo cp include/cudnn.h /usr/local/cuda-7.5/include/ sudo ldconfig

```

我想在其他文件夹太 sudo cp lib* /usr/local/cuda/lib64/ sudo cp include/cudnn.h /usr/local/cuda/include/ sudo ldconfig

+0

您可以使用LD_DEBUG查找加载哪个版本的cudnn库以及该库的位置?如果它仍然是旧库,则可以使用v5.1更新它。 –

+0

同样的问题在这里 - 到目前为止运气还不错?我试过使用LD_DEBUG,但是我一定在做错事。我运行'LD_DEBUG = all cat',然后在另一个窗口中运行TensorFlow代码,但没有显示出来。 –

+0

这是我的一个主要问题。我必须确保我重新安装了所有东西。您是在AWS还是您的本地计算机上运行? – Ritchie

回答

7

有什么意思在这里一个很好的解释 - What does the error: `Loaded runtime CuDNN library: 5005 but source was compiled with 5103` mean?

简短的回答是,你必须CuDNN 5.0,但你应该安装CuDNN 5.1

看起来这就是你正在尝试做的事情。它的工作对我来说,只需按照指示在这里 - https://www.tensorflow.org/get_started/os_setup#optional_install_cuda_gpus_on_linux

在此之前,我查的/usr/local/cuda/include/cudnn.h内容确实也有这些线向上方表明它是5.0.5版

#define CUDNN_MAJOR  5 
#define CUDNN_MINOR  0 
#define CUDNN_PATCHLEVEL 5 

如果您的/usr/local/cuda/include/cudnn.h已经是5.1,那么另一个目录中的另一个CuDNN正在被引用。我在我的.bashrc中有以下内容 - 也许尝试添加此内容或检查Tensorflow instructions要添加的内容。

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64" 
export CUDA_HOME="/usr/local/cuda"