Exhausted GPU memory with batch size of 1

Question

wmcnally opened this issue 3 years ago · 2 comments

I have 12GB of GPU memory (Titan Xp) and I ran out of memory using a batch size of 1 (when running valid.py). Is this expected? Thanks.

Answer 1 · 2021-09-28T01:45:21.000Z

Hi, I think the code can run on Titan Xp. You can check other settings.

Answer 2 · 2021-09-30T19:28:41.000Z

Not sure what happened... It appears to be working now and it is only consuming ~3.6 GB of memory.