RuntimeError: CUDA out of memory. Tried to allocate 1.32 GiB (GPU 0; 7.92 GiB total capacity; 5.68 GiB already allocated; 1.22 GiB free; 192.30 MiB cached)

[xuyuhui666]

My commands :
python train.py -model_type img -data data/demo -save_model demo-model -gpu_ranks 0 -batch_size 4 -learning_rate 0.1 -word_vec_size 80 -encoder_type brnn -image_channel_size 3

My GPU breaks while running to the 10000 steps of 100000

My batch_size sets 4 ,as 20 defined can not run

[liminghao1630]

What size is your GPU memory? Try to use a smaller batch size.