美文网首页
Could not create cudnn handle: C

Could not create cudnn handle: C

作者: 热爱生活H_Li | 来源:发表于2021-04-19 08:28 被阅读0次

    cudnn已经读进去了


    image.png

    加如下代码即可

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

    引用:https://blog.csdn.net/pkuyjxu/article/details/89402298

    相关文章

      网友评论

          本文标题:Could not create cudnn handle: C

          本文链接:https://www.haomeiwen.com/subject/emlulltx.html