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

本人环境:
anaconda 3.6
CUDA 10.0
cudnn10.0
tensorflow-gpu 2.0
做卷积神经网络时报错

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

在确保各个版本都适配的情况下,考虑显存设置问题
设置为仅在需要时申请显存
解决办法:在代码中加上:

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)

你可能感兴趣的:(人工智能,神经网络,深度学习,tensorflow)