上传者: 38733245
|
上传时间: 2022-04-20 22:41:40
|
文件大小: 71KB
|
文件类型: PDF
Tensorflow2.0 之Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED问题
问题描述:
在tensorflow2.0的学习过程中,遇到了Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED,发现这个问题是我在调用kseras模块的卷积类前向传播时触发的,之前的全连接层都没有出现过类似的问题。分析问题应该是gpu没有正确的分配所导致的。
import tensorflow as tf
from tensorflow.keras import layers
x =