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)
上一篇: deepin安装mysql5.6
下一篇: 请问,你这里还需要清洁工吗?收废品的也行