RuntimeError: CUDA out of memory
shin7 opened this issue · 1 comments
shin7 commented
Hello, I have met a problem when training the model as follow:
RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 983.69 MiB total capacity; 209.53 MiB already allocated; 69.56 MiB free; 6.47 MiB cached)
could you give me any solutions?
Thank you!
kei6 commented
It seems your device is not strong enough to train this model. 1G of GPU is really small.
I'm using 16Gb GPU and this model uses its all.