Pytorch运行bug

在代码书写没有bug的情况下，Pytorch竟然报了这个错：

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

起因是跑ImageNet时，batchsize设得过大了，超过256就报错了，比如512和1024都不行，就连多卡并行也不行；似乎是cuDNN的版本问题，目前仍没有解决

环境:

　　torch1.5.0;

　　cuda 10.2