Down-to-the-Last-Detail-Virtual-Try-on-with-Detail-Carving icon indicating copy to clipboard operation
Down-to-the-Last-Detail-Virtual-Try-on-with-Detail-Carving copied to clipboard

Published 20 hours ago • verified publisher icon JDAI-CV

Cuda Memory Error

Open atulnagane45 opened this issue 3 years ago • 1 comments

I am getting this error RuntimeError: CUDA out of memory. Tried to allocate 648.00 MiB (GPU 0; 3.82 GiB total capacity; 1.57 GiB already allocated; 665.06 MiB free; 1.98 GiB reserved in total by PyTorch)

My hardware is Geforce GTX1650 4GB dedicated memory I tried this on google colab too.

I already tried : Batch size from 128 to 2 Cleared GPU cache.

I think it need to be fixed from code. Some variables unnecessarily taking too much space. Please help me. Thanks in advance

atulnagane45 avatar Mar 11 '21 05:03 atulnagane45

Solved on google collab with batch size 2. What is minimum hardware requirement?

atulnagane45 avatar Mar 11 '21 07:03 atulnagane45