SegNet-Tutorial icon indicating copy to clipboard operation
SegNet-Tutorial copied to clipboard

Out of memory error

Open chinnikrishna55 opened this issue 7 years ago • 1 comments

I have 4GB of GPU memory and getting out of memory error with just batch_size of 2

F1215 16:35:47.650579 5231 syncedmem.cpp:51] Check failed: error == cudaSuccess (2 vs. 0) out of memory *** Check failure stack trace: *** @ 0x7f5436877daa (unknown) @ 0x7f5436877ce4 (unknown) @ 0x7f54368776e6 (unknown) @ 0x7f543687a687 (unknown) @ 0x7f5436ccfb3b caffe::SyncedMemory::mutable_gpu_data() @ 0x7f5436cd2222 caffe::Blob<>::mutable_gpu_data() @ 0x7f5436cf4a8f caffe::BNLayer<>::Forward_gpu() @ 0x7f5436bdc349 caffe::Net<>::ForwardFromTo() @ 0x7f5436bdc777 caffe::Net<>::ForwardPrefilled() @ 0x7f5436ce6f85 caffe::Solver<>::Step() @ 0x7f5436ce78bf caffe::Solver<>::Solve() @ 0x406676 train() @ 0x404bb1 main @ 0x7f5435d89f45 (unknown) @ 0x40515d (unknown) @ (nil) (unknown) Aborted (core dumped)

chinnikrishna55 avatar Dec 15 '16 21:12 chinnikrishna55

decrease your batch size.

acgtyrant avatar Dec 28 '16 03:12 acgtyrant