矩池云 · 2021年01月29日

解决This is probably because cuDNN failed to initial

image.png

报错内容

tensorflow.python.framework.errors_impl.UnknownError:  Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

网上文章都是说cudnn版本不匹配,但是查了之后发现不是这个问题。

image.png

https://www.tensorflow.org/in...

然后看到了一篇博客说到可能是因为显存不足,导致的。

限制GPU的使用率

方法一

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.85  #占用85%显存
session = tf.Session(config=config)

方法二

gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.85)
config=tf.ConfigProto(gpu_options=gpu_options)
session = tf.Session(config=config)

参考文章

矩池云中Tensorflow指定GPU及GPU显存设置

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

推荐阅读
关注数
1073
内容数
18
矩池云国内领先的GPU云共享平台,提供高性能的GPU租赁服务,助力人工智能、影视渲染行业。
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息