This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
在跑基于tensorflow的FCN时碰到这个问题,网上翻提示的时候发现提示Loaded runtime CuDNN library: 7.0.5 but source was compiled with: 7.1.4.
截图:


主要原因是因为tensorflow版本过高。
一开始跑代码的的时候实在tensorflow1.12下跑的。后来又新建的一个tensorflow1.10的环境,再次跑的时候成功运行。
本文详细解析了在使用基于tensorflow的全卷积网络(FCN)时遇到的CuDNN版本不匹配错误。主要原因是源代码编译使用的CuDNN版本与运行时加载的版本不符,通过降低tensorflow版本至1.10,成功解决了该问题。

被折叠的 条评论
为什么被折叠?



