ESRGAN icon indicating copy to clipboard operation
ESRGAN copied to clipboard
verified publisher icon xinntao

RuntimeError: CUDA out of memory.

Open overthrowrobotics opened this issue 6 years ago • 1 comments

Tried to allocate 7.91 GiB (GPU 0; 10.73 GiB total capacity; 2.56 GiB already allocated; 6.63 GiB free; 761.56 MiB cached)

I'm using two 2080Ti with 11GB each and have CUDA_VISIBLE_DEVICES set to 0,1. Can the model be evaluated with both the memory from both GPUs? Input file is 1920x1080. Maybe I just need to split the file into smaller pieces and then recombine. Thoughts?

overthrowrobotics avatar Jun 10 '19 21:06 overthrowrobotics

Your input is really large (1920x1080). Yes, you need to split the image into smaller pieces and then recombine.

xinntao avatar Jun 12 '19 16:06 xinntao