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

     
     
     
  • 相关阅读:
    jQuery教程
    AJAX请求 $.ajax方法的使用
    smarty block_function
    smarty
    位运算版本的交换两数
    提取字符串中的数字
    vue地址插件多级联动自适应 + github地址
    vue插件
    网页title旁边的小图片
    二十三种设计模式[14]
  • 原文地址:https://www.cnblogs.com/Renyi-Fan/p/13458683.html
Copyright © 2011-2022 走看看