规格:
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
。
答案 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.5
和libcudnn.so.5.1.10
。