run out of memory

[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.

[MeioJane]

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

[Eli-Goldberg]

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).