CUDA out of memory

[snowymo]

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.