Cuda out of memory even batch size is reduced to 2

herbert-chen / tianchi_lvcai

天池竞赛Baseline分享（2018广东工业智造大数据创新大赛——智能算法赛，线上0.921无调参）

255 stars 98 forks source link

Cuda out of memory even batch size is reduced to 2 #9

Open kailiyumit opened 4 years ago

kailiyumit commented 4 years ago

I still have the cuda out of memory issue after I reduce the batch size:

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.25 GiB already allocated; 3.55 MiB free; 1.27 GiB reserved in total by PyTorch)

Any suggestions?