zoukankan      html  css  js  c++  java
  • “Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法

    “Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法

    一、总结

    一句话总结:

    错误原因是我的图像比较大,消耗GPU资源较多。但我的显卡(RTX2060)显存只有6GB,所以会出现这个错误。
    解决方法就是加一段对GPU进行按需分配代码
    from tensorflow.compat.v1 import ConfigProto
    from tensorflow.compat.v1 import InteractiveSession
    
    config = ConfigProto()
    config.gpu_options.allow_growth = True
    session = InteractiveSession(config=config)

    二、“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法

    转自或参考:“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法
    https://blog.csdn.net/tsyccnh/article/details/102938368

    最近在使用TF2.0。运行程序出现以下错误。

    Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
    

    一开始怀疑是CUDA和CuDNN配置错误(要求版本匹配)。反复试验后,还是有这个错误。
    最后发现可能是GPU内存不足造成的。需要在程序前加以下一段代码:

    from tensorflow.compat.v1 import ConfigProto
    from tensorflow.compat.v1 import InteractiveSession
    
    config = ConfigProto()
    config.gpu_options.allow_growth = True
    session = InteractiveSession(config=config)
    

    意思是对GPU进行按需分配。
    主要原因是我的图像比较大,消耗GPU资源较多。但我的显卡(RTX2060)显存只有6GB,所以会出现这个错误。这个错误提示有很大的误导性,让人一直纠结CUDA和CuDNN的版本问题。故在此立贴,以免后人重蹈覆辙。


    参考:

    https://github.com/tensorflow/tensorflow/issues/24828

     
     
     
  • 相关阅读:
    二叉树的下一个节点
    二叉树的对称
    CString,string和char*
    二叉平衡树
    二叉树的深度
    必应首页图片下载
    Git报错:fatal: remote origin already exists.
    sublime text3 自定义代码片段
    atom自定义C++代码片段
    vscode 自定义代码片段(snippets)
  • 原文地址:https://www.cnblogs.com/Renyi-Fan/p/13458683.html
Copyright © 2011-2022 走看看