CUDNN_STATUS_NOT_SUPPORTED error. pytorch and cuda version?

[JudyYe]

hi, I get the following error even at the lowest resolution.

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

It somehow changes the input to non-backward-able after F.conv2d at this line, called by rgb[0].

Some existing issues points me to cudnn bugs and the workaround is the have the `correct' version. (e.g. pytorch 1.4.0 may produce this issue while 1.1 don't) . Would you mind sharing with me your pytorch and cuda version? Thank you.