SAN_pytorch icon indicating copy to clipboard operation
SAN_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon Yukariin

CUDA out of memory

Open CR7forMadrid opened this issue 2 years ago • 1 comments

Why CUDA out of memory .Tried to allocate 8.38GiB (GPU 0; 10.92 GiB total capacity; 8.69 GiB already allocated; 1.22GiB free;33.00 MiB cached) Why I set batch_size=1，but still CUDA out of memory.what is the reason

CR7forMadrid avatar Nov 01 '22 01:11 CR7forMadrid

What are you trying to do exactly? Your 11GB GPU should be fine for inference (using test.py), even with batch_size=16 As for training, I used batch_size=16 with patch size of 48x48 on T4

Yukariin avatar Nov 01 '22 19:11 Yukariin