日志提示“CUDNN_STATUS_NOT_SUPPORTED. ”

问题现象

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

原因分析

出现该问题的可能原因如下：

数据输入不连续，cuDNN不支持的类型。

处理方法

1. 禁用cuDNN，在训练前加入如下代码。

   torch.backends.cudnn.enabled = False

2. 将输入数据转换成contiguous。

   images = images.cuda()  
   images = images.permute(0, 3, 1, 2).contigous()

建议与总结