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

报错信息提示:

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)

即可解决。

©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页