tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.

解决报错:tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.

错误

tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm._第1张图片

解决办法

升级CuDNN

根据输出窗口的提示

tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm._第2张图片

这里说明需要更高版本的CuDNN

以我为例这里提示我,我的环境中的CuDNN是7.4.1,不满足环境需求。之后我将CuDNN升级到7.6.5,将问题解决。

如何升级?可以参考其他博主的文章。

你可能感兴趣的:(BUG,tensorflow)