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

posted @ 2020-09-09 09:07  marsggbo  阅读(4742)  评论(0编辑  收藏  举报