openai / guided-diffusion

MIT License
6.03k stars 803 forks source link

out of memory in FID evaluation #82

Closed forever208 closed 1 year ago

forever208 commented 1 year ago

failed to allocate 64.00M (67108864 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory

Does anyone have a problem using GPU when doing evaluation?

ridhimagarg commented 1 year ago

I think you have to use better GPU, but for me I didn't face use of memory while evaluating.

What is the configuration of your GPU?

forever208 commented 1 year ago

I roughly know that it was the problem with the CPU memory, thanks @ridhimagarg