JingyunLiang / VRT

VRT: A Video Restoration Transformer (official repository)
https://arxiv.org/abs/2201.12288
Other
1.37k stars 130 forks source link

How much memory about GPUs? #56

Open focus1024-wind opened 1 year ago

focus1024-wind commented 1 year ago

When I train on my host, it issue the error of out of memory. My host is configured with 4 x 24GB NVIDA3090 GPUs. I see your file that say you use 8 A100 GPUs to train the task. But there are two configurations about A100, so i want to konw concrete configuration about you GPUs or how much memory should I need. Looking forward to your help, thanks.

jqtangust commented 1 year ago

I also used 3090 too, but it is still out of memory.

Shengqi77 commented 1 year ago

I used 8 3090,but it is also out of memory.

yyhtbs-yye commented 7 months ago

It needs about 32GB