规格:
Ubuntu 16.04
tensorflow 1.1.0
NVIDIA 850m 4GB
CUDA 8
CUDNN 6
我创建了一个软链接ln -s libcudnn.so.6.* libcudnn.so.5
建议here。另外我得到一个错误,ImportError: libcudnn.5: cannot open shared object file: No such file or directory
。
尝试安装不同的cuDNN,如果你使用CUDA 8你可以尝试cuDNN 5.1,如下所述:https://softwaremill.com/how-to-upgrade-aws-deep-learning-ami-to-tensorflow-1-1-0/(仅查看cuDNN部分)
创建一个软链接ln -s libcudnn.so.6.* libcudnn.so.5
没有任何效果。
只需从cuDNN Download下载cuDNN v5.1并安装它。
之后,它将自动创建libcudnn.so.5
和libcudnn.so.5.1.10
。