在GPU运行TensorFlow UNET模型时出现CUDNN_STATUS_EXECUTION_FAILED错误求助

标题 #

二级标题 #

三级标题 #

四级标题

列表项 #

• 基础列表项1
• 基础列表项2
  • 嵌套子列表项1
  • 嵌套子列表项2

文本强调 #

斜体强调的内容
粗体强调的内容

代码或命令 #

单行命令示例：nvcc --version
多行代码示例：

from tensorflow.keras import layers
def unet_block(inputs, filters):
    x = layers.Conv2D(filters, 3, padding='same')(inputs)
    x = layers.BatchNormalization()(x)
    x = layers.Activation('relu')(x)
    return x

引用文本 #

CUDNN_STATUS_EXECUTION_FAILED
in tensorflow/compiler/xla/stream_executor/cuda/cuda_dnn.cc(5750): 'cudnnBatchNormalizationForwardTrainingEx(...)'

链接 #

TensorFlow 2.11.0 官方文档

图片 #

Quadro P620 显卡示意图

内容的提问来源于stack exchange，提问作者user3323171