出现RuntimeError: cuDNN error: CUD

在Pytorch模型的forward函数中报错：

RuntimeError: cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input

报错的原因是，一个batch的数据量过大。解决方案是减小batch size。

本地的CUDA版本是10.0，cudnn版本为7.6.3。同样的问题在其他版本上也会出现：https://discuss.pytorch.org/t/cudnn-status-not-supported-for-large-matrix-input/10986。