zoukankan      html  css  js  c++  java
  • 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

    安装完成即可!

  • 相关阅读:
    spring reference
    Connector for Python
    LDAP
    REST
    java利用泛型实现不同类型可变参数
    java细节知识
    事务隔离的级别
    servlet cdi注入
    session and cookie简析
    CORS’s source, principle and implementation
  • 原文地址:https://www.cnblogs.com/aidenzdly/p/10369149.html
Copyright © 2011-2022 走看看