在代码书写没有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