Issue for Cuda

Open ThanhPham1987 opened this issue 6 years ago • 3 comments

Hi biu, When i run command line for training. I'm facing this issue: RuntimeError: CUDA out of memory. Tried to allocate 200.00 MiB (GPU 0; 3.94 GiB total capacity; 2.79 GiB already allocated; 208.50 MiB free; 28.71 MiB cached).

Please give me some advice.

Best regards, PeterPham

Nov 29 '19 04:11 ThanhPham1987

Same question,it seems like that the train.py is runned by many times,and the gpu memory is filled. did you solve the question?thanks!

Oct 17 '22 14:10 DennisJcy

@DennisJcy Change the batch size when you train model

Dec 15 '22 04:12 ThanhPham1987

@ThanhPham1987 thank you for your reply!🥸

Dec 17 '22 04:12 DennisJcy