CUDA out of memory #10
Description
Hi, I wonder what is the computer's graphic memory when training the data? I received the error msg:
RuntimeError: CUDA out of memory. Tried to allocate 9.00 MiB (GPU 0; 8.00 GiB total capacity; 641.54 MiB already allocated; 8.70 MiB free; 474.50 KiB cached)
I keep printing nvidia-smi and found that it reached ~7GB and finally reported the error because I only have 8 GB memory.
I wonder how much memory is required? If it does require more than 8 GB memory, any suggestion for me to change the code? Thanks a lot.