爆显存(CUDA OOM)什么原因?

问题 爆显存(CUDA OOM)什么原因? 原因 A:如果执行程序报错OOM显存不足,那么最简单的调试办法是设置batch size=1然后逐步增大,观察跑到多大bs时OOM,以此为依据再决定升配成多…

close
arrow_upward