donut icon indicating copy to clipboard operation
donut copied to clipboard

How much gpu memory size does single 1280,960 size photo need

Open Mrharry1997 opened this issue 2 years ago • 2 comments

I tried to run donut-base in 2080ti, batch set is 1. But it didn't work, and it looked like it's because the small gpu memory. So I want to ask has anyone try to run it on 2080ti or how much gpu memory size does single 1280,960 size photo need

Mrharry1997 avatar Aug 17 '22 13:08 Mrharry1997

I was able to run the 1280x960 in a 16gb GPU t4 system , so may be you can try running it in cpu? or colab

TheSeriousProgrammer avatar Aug 19 '22 09:08 TheSeriousProgrammer

It takes 10-12GB of GPU RAM to train with an image size of 1280x960 and a batch size of 1. Here is my log when training the model image

maycuatroi avatar Aug 31 '22 17:08 maycuatroi

Close this issue since it seems to be resolved :) Feel free to reopen this or open another issue if you have anything new for sharing or debugging.

gwkrsrch avatar Nov 11 '22 07:11 gwkrsrch