E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC
本人环境:anaconda 3.6CUDA 10.0cudnn10.0tensorflow-gpu 2.0做卷积神经网络时报错E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILE在确保各个版本都适配的情况下,考虑显存设置问题设置为仅在需要时申请显存解决办法:在代码中加上:import tensorflow as tfconfig
原创
2020-06-03 11:50:19 ·
1904 阅读 ·
5 评论