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或者减少输入数据大小,亲测有效。

微信公众号:AutoML机器学习
MARSGGBO原创
如有意合作或学术讨论欢迎私戳联系~
邮箱:marsggbo@foxmail.com

原文地址:https://www.cnblogs.com/marsggbo/p/13636982.html