2020年3月1日
模型训练错误：Failed to get convolution algorithm，This is probably because cuDNN failed to initialize.
摘要： 一、问题： 最近在利用keras训练模型的时候出现以下错误，开始一看以为是cuDNN的版本不对，更换后还是没有效果，后面在网上查找后发现是显存的问题，更改为动态分配内存就可以了。 二、解决方法： 在训练的脚本开头添加以下代码进行动态分配内存，然后重新运行就可以。 import tensorflow 阅读全文
posted @ 2020-03-01 09:35 半夜打老虎 阅读(12918) 评论(0) 推荐(1)