Tensorflow错误:加载运行时CuDNN库:6021(兼容版本6000)但源代码是使用5110编译的(兼容版本5100)

时间:2017-06-13 13:31:33

标签: tensorflow cudnn

规格:

Ubuntu 16.04
tensorflow 1.1.0
NVIDIA 850m 4GB
CUDA 8
CUDNN 6

我已根据建议here创建了一个软链接ln -s libcudnn.so.6.* libcudnn.so.5。否则我收到错误ImportError: libcudnn.5: cannot open shared object file: No such file or directory

这是usr/local/cuda/lib64的快照 enter image description here 我该怎么做才能解决错误?

2 个答案:

答案 0 :(得分:3)

尝试安装不同的cuDNN,如果你使用CUDA 8,你可以尝试cuDNN 5.1,如下所述:https://softwaremill.com/how-to-upgrade-aws-deep-learning-ami-to-tensorflow-1-1-0/(仅查看cuDNN部分)

答案 1 :(得分:0)

创建软链接ln -s libcudnn.so.6.* libcudnn.so.5无效。

只需从cuDNN Download下载cuDNN v5.1并安装即可。

之后,它会自动创建libcudnn.so.5libcudnn.so.5.1.10

enter image description here