Created by: ghost

Hello, I have a problem.

I successfully run ResNet50 with batch_size 64 on Titan V card (12gb). Even with batch_size 96 it works well, but with batch_size 128 it stops working.

I tried shrinking FLAGS_fraction_of_gpu_memory_to_use from 0.98 (default) even to 0.1 but it doesn't work...

Cannot malloc 49.0002 MB GPU memory. Please shrink FLAGS_fraction_of_gpu_memory_to_use or FLAGS_initial_gpu_memory_in_mb or FLAGS_reallocate_gpu_memory_in_mbenvironment variable to a lower value. Current FLAGS_fraction_of_gpu_memory_to_use value is 0.1. Current FLAGS_initial_gpu_memory_in_mb value is 0. Current FLAGS_reallocate_gpu_memory_in_mb value is 0.

What's the problem here? Can you help me with that?