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

    安装完成即可!

  • 相关阅读:
    人脸识别员工考勤系统
    栈和队列
    线性表
    C语言博客作业02--循环结构
    课程设计-个人博客
    C博客作业02--循环结构
    博客作业--函数
    c博客作业
    联系方式
    专业特长
  • 原文地址:https://www.cnblogs.com/aidenzdly/p/10369149.html
Copyright © 2011-2022 走看看