天天看点

Pytorch cuDNN error: CUDNN_STATUS_NOT_SUPPORTED.解决办法

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

如果遇到这个问题,原因可能是计算量太大,你可以尝试减小batch size或者减少输入数据大小,亲测有效。

继续阅读