deep-image-prior icon indicating copy to clipboard operation
deep-image-prior copied to clipboard

Published 20 hours ago • verified publisher icon DmitryUlyanov

Out of memory

Open baobabKoodaa opened this issue 6 years ago • 2 comments

RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1512387374934/work/torch/lib/THC/generic/THCStorage.cu:58

After googling a bit it seems like my GPU does not have enough memory to run the examples? Would it be possible to get like a ballpark estimate on what kind of machine is required to run the notebooks?

baobabKoodaa avatar Dec 06 '17 17:12 baobabKoodaa

We used 12GB GPU. You can try to reduce the number of filters in the network or image resolution for example, there is a chance it still works nicely after such changes.

DmitryUlyanov avatar Dec 06 '17 21:12 DmitryUlyanov

It might be worth looking into the pytorch variable allocation, which can have a big impact on required GPU memory[1]. But I'm not really seeing any major issues in this code.

turbo avatar Dec 06 '17 22:12 turbo