欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-05-26 23:43:33
...

本人环境:
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)