search

CVMI-Lab / UHDM

(ECCV2022) This is the official PyTorch implementation of ECCV2022 paper: Towards Efficient and Scale-Robust Ultra-High-Definition Image Demoireing
Apache License 2.0
191 stars 25 forks source link

Have you tested how much video memory will be used when using the GPU? #6

Closed 1362802590 closed 1 year ago

1362802590 commented 1 year ago

HI,Have you tested how much video memory will be used when using the GPU?

XinYu-Andy commented 1 year ago

What do you mean video?

HI,Have you tested how much video memory will be used when using the GPU?

1362802590 commented 1 year ago

video memory 显存

XinYu-Andy commented 1 year ago

I remember it would cost around 9000MiB GPU memory when testing a 4K image. As for the training, around 11000MiB for 2 batch_size * (768,768,3) resolution image. Not sure for this, just a rough estimation.

1362802590 commented 1 year ago

ok,thanks