RuntimeError: CUDA error: out of memory

Question

RuntimeError: CUDA error: out of memory

Abel1802 opened this issue 5 years ago · 5 comments

RuntimeError Traceback (most recent call last)
in
11
12 if USE_CUDA:
---> 13 input_labels = input_labels.cuda()
14 pos_labels = pos_labels.cuda()
15 neg_labels = neg_labels.cuda()

Answer 1 · 2019-11-17T06:19:18.000Z

Where is this error coming from? Can you reduce the batch size or change the hyper parameters to make the model smaller?

Answer 2 · 2019-11-17T06:38:55.000Z

thanks for your reply！ I followed the code on your tutorial(no modification).But it went wrong while training the model(in the follow picture).I first use cuda with gpu on my lab. So I don't how to resolve this bug. At 2019-11-17 14:19:19, "Zewei Chu" <notifications@github.com> wrote: Where is this error coming from? Can you reduce the batch size or change the hyper parameters to make the model smaller? — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Answer 3 · 2019-11-17T06:39:46.000Z

At 2019-11-17 14:19:19, "Zewei Chu" <notifications@github.com> wrote: Where is this error coming from? Can you reduce the batch size or change the hyper parameters to make the model smaller? — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Answer 4 · 2019-11-20T08:16:31.000Z

Where is the picture you mentioned in your reply?

Answer 5 · 2019-11-21T15:17:02.000Z

I have solved this problem.But I am also grateful to you! Have a happy life! Good-bye! At 2019-11-20 16:16:31, "Zewei Chu" <notifications@github.com> wrote: Where is the picture you mentioned in your reply? — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.