RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED解决方案

问题描述：已经跑通的代码换个电脑出现报错

报错代码：

loss.backward()

 cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

问题原因及解决方案；

1.cudnn与cuda或pytorch未对应，请自查对应关系重新安装（大概率不是这个）

2.程序需要的显存过大导致报错，俗称爆显存，调小batch-size进行尝试

3.报错代码是loss.backward()，请查看代码中的类别数与数据集的类别数是否正确对应

解决方案2、3的报错代码都有可能是loss.backward()........请仔细查验代码~