Keras在MNIST实现LeNet-5模型训练时的错误？

当使用Keras API 训练模型时，训练时报错？

UnknownError (see above for traceback): Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

在运行手写体数字识别的过程的中报错如上。

 

 

 

那么该如何解决这个问题呢？

其实很简单：

原因是兼容问题，tensorflow的版本过高而和安装的cudnn（我的是7.0）和cuda(我的是9.0)不匹配，可能你使用的是tensorflow的最新的gpu版本，比如说我当时用的就是tensorflow和tensorflow-gpu 1.12版本

就会出现这个错误，后来降低tensorflow版本就可以了

 

我用的是windows    直接 pip install tensorflow==1.9.0           

   　　　　　　   　　     pip install tensorflow-gpu==1.9.0

安装完成即可！

转载于:https://www.cnblogs.com/aidenzdly/p/10369149.html