CUDA out of memory

Open justinzhang528 opened this issue 2 years ago • 5 comments

進行訓練的時候出現=》RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 10.92 GiB total capacity; 10.17 GiB already allocated; 99.81 MiB free; 10.20 GiB reserved in total by PyTorch) 我的GPU配置： NVIDIA GeForce GTX 1080 Ti 11264 MB 我有嘗試把batch_sizej降低至1，還是未能解決，不知是否還有其他方法可以解決問題

Feb 20 '23 04:02 justinzhang528

first try with batch size 1, if it doesn't work try reducing dataset size

Sep 15 '23 18:09 karelbecerra

现在解决了吗

Jan 13 '24 08:01 nchzzDFTBA

你好，请问你解决了吗

Jan 31 '24 11:01 chjwhere

沒有解決喔，後來選擇放棄，直接用其它模型

Jan 31 '24 11:01 justinzhang528

你好，我也遇到相关问题无法解决。请问你后来使用的是什么模型，能否给个参考。谢谢！

Feb 03 '24 09:02 chjwhere