RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR #27588
错误的提示信息如下:先说解决方法:1.将batchsize继续调小,直到不出现问题2.添加如下代码:torch.backends.cudnn.enabled = False原因分析:可能是cudnn的一个bug,由Conv3d layer引起的参考:https://github.com/pytorch/pytorch/issues/27588...
文章共129字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
错误的提示信息如下:
先说解决方法:
1.将batchsize继续调小,直到不出现问题
2.添加如下代码:
torch.backends.cudnn.enabled = False
原因分析:
可能是cudnn的一个bug,由Conv3d layer引起的
参考:https://github.com/pytorch/pytorch/issues/27588
更多推荐
已为社区贡献1条内容
所有评论(0)