nashory/DeLF-pytorch

CUDA out of memory when running notebook/visualize.ipynb

thoang3 opened this issue · 2 comments

Hi @nashory and everyone,

I have the following error when running get_result(myfeeder, query) in the visualize.ipynb notebook. Could you show me how to fix this? I've tried to reduce workers = 1 to no avail. Thank you in advance!

RuntimeError: CUDA out of memory. Tried to allocate 374.00 MiB (GPU 0; 10.73 GiB total capacity; 9.57 GiB already allocated; 330.62 MiB free; 66.62 MiB cached)

maybe you could resize image size

have you the question
1.no module named feeder
2.no module named matcher