规格 :
Ubuntu 16.04
tensorflow 1.1.0
NVIDIA 850m 4GB
CUDA 8
CUDNN 6
我已经按照这里ln -s libcudnn.so.6.* libcudnn.so.5
的建议创建了一个软链接。否则我得到一个错误,.ImportError: libcudnn.5: cannot open shared object file: No such file or directory
规格 :
Ubuntu 16.04
tensorflow 1.1.0
NVIDIA 850m 4GB
CUDA 8
CUDNN 6
我已经按照这里ln -s libcudnn.so.6.* libcudnn.so.5
的建议创建了一个软链接。否则我得到一个错误,.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 下载 cuDNN v5.1 下载并安装它。
libcudnn.so.5
之后,它将libcudnn.so.5.1.10
自动创建。