SegNet-Tutorial
SegNet-Tutorial copied to clipboard
Out of memory error
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)
decrease your batch size.