zl程序教程

您现在的位置是:首页 >  其它

当前栏目

RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

to Find unable run valid ALGORITHM cudnn RuntimeError
2023-09-11 14:20:14 时间

在这里插入图片描述
参考这里
https://stackoverflow.com/questions/61467751/unable-to-find-a-valid-cudnn-algorithm-to-run-convolution
说是降低batch_size可能有用
在这里插入图片描述
查看显存
发现快到了顶了,说明确实可能是显存的问题
在这里插入图片描述
调小之后解决了