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
原创
2020-06-03 11:50:19 ·
1808 阅读 ·
5 评论
关注
分享
扫一扫