IuException: CUDA Error: out of memory

[grandyang]

Hi Patrick

I tried the example command line "./stereo_img --im0 ../data/im0.png --im1 ../data/im1.png", but got the following error:
terminate called after throwing an instance of 'IuCudaException'
what(): IuException: CUDA Error: out of memory
where: /home/xiaochenyang/cnn-crf-stereo/dependencies/imageutilities/include/iu/iucore/lineardevicememory.h | LinearDeviceMemory:60
Aborted (core dumped)

I thought maybe my GPU (nvs 310) is low-level that does not have enough memory to run the code. However, I was wondering do you have any parameter that can adjust batch size so that it can run on low memory GPU. By the way, is there any memory limit to run this code?

Thanks a lot!

[pknoebelreiter]

Hi,
for running the code you will need ~3GB GPU memory for the 3 layer CNN and ~5GB GPU memory for the 7 layer CNN. The batch size is always 1 for this implementation. If I am not wrong your card has only 1GB memory, which is not enough. The only thing you could try to reduce the memory footprint is reducing the input image size and/or using less disparity steps, e.g. 64 instead of 128.