Ftensorflow/stream_executor/cuda/cuda_dnn.cc:522]检查失败

Question

我正在张量流中训练我的模型。经过一些迭代后，我得到以下异常：

F tensorflow/stream_executor/cuda/cuda_dnn.cc:522] 检查失败： cudnnSetTensorNdDescriptor(handle_.get(), elem_type, nd, dims.data(), strides.data()) == CUDNN_STATUS_SUCCESS (3 vs. 0)batch_descriptor ：{计数：1 feature_map_count：1088 空间：63 0 value_min：0.000000 value_max：0.000000 布局：BatchDepthYX}

问题不在于 cuda 库，因为相同的代码适用于其他数据集。这可能是什么原因呢？