【tensorflow 错误解决】Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

东华果汁哥

于 2020-03-13 12:30:07 发布

阅读量1.2w

点赞数 32

分类专栏：深度学习--tensorflow

本文链接：https://blog.csdn.net/u013421629/article/details/104838307

版权

深度学习--tensorflow 专栏收录该内容

41 篇文章

订阅专栏

报错信息提示：

E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

原因分析：

GPU显存设置有问题，需要设置为仅在需要时申请显存。

import tensorflow as tf
config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))
sess = tf.compat.v1.Session(config=config)

即可解决。