show_attend_and_tell_pytorch icon indicating copy to clipboard operation
show_attend_and_tell_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon parksunwoo

CUDA Out of Memory issue

Open KUNAL1612 opened this issue 5 years ago • 1 comments

Running python main.py with checkpoint None throws up error as RuntimeError: CUDA out of memory. Tried to allocate 6.12 MiB (GPU 0; 3.95 GiB total capacity; 2.70 GiB already allocated; 19.81 MiB free; 2.17 MiB cached)

KUNAL1612 avatar Jul 24 '19 22:07 KUNAL1612

You have only 4GB space in GPU. The network might need more than that. Either train it on a system with more GPU space or on google colab.

vinits5 avatar Jan 06 '20 00:01 vinits5