CHR icon indicating copy to clipboard operation
CHR copied to clipboard

run out of memory

Open xiying-wei opened this issue 6 years ago • 2 comments

When I run the program,Instead of using the GPU, the memory utilization keeps increasing until it is 100%.How to solve this problem?Thank you.

xiying-wei avatar Oct 13 '19 03:10 xiying-wei

When I run the program,Instead of using the GPU, the memory utilization keeps increasing until it is 100%.How to solve this problem?Thank you.

I train or test using the GPU, I do not have a try without GPU

MeioJane avatar Nov 25 '19 07:11 MeioJane

I just lowered the batch size to around 8 and then it used around 4GB of VRAM. On a 16GB card you can up that number to as high as 80 (but for some reason training will not be quicker).

Eli-Goldberg avatar Jan 11 '23 22:01 Eli-Goldberg