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

溴麝香草酚蓝水溶液

于 2020-06-03 11:50:19 发布

阅读量1.9k

点赞数 3

分类专栏：人工智能文章标签：神经网络深度学习 tensorflow

本文链接：https://blog.csdn.net/weixin_42299634/article/details/106521189

版权

在使用CUDA 10.0、cudnn 10.0和tensorflow-gpu 2.0进行卷积神经网络训练时，遇到错误'E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC'。该问题可能由于显存设置导致。解决方案是在代码中动态分配显存，只在需要时申请。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

最低0.47元/天解锁文章