how much gpu should i need to train the model?
MrCrazyCrab opened this issue · 0 comments
MrCrazyCrab commented
RuntimeError: CUDA out of memory. Tried to allocate 93.50 GiB (GPU 0; 7.79 GiB total capacity; 31.26 MiB already allocated; 6.08 GiB free; 12.74 MiB cached) (malloc at /pytorch/c10/cuda/CUDACachingAllocator.cpp:267)
i have change the bs to 1, but i still meet the problem that the cyda out of memory.