CUDA out of memory

[justinzhang528]

進行訓練的時候出現=》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，還是未能解決，不知是否還有其他方法可以解決問題

[karelbecerra]

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

[nchzzDFTBA]

现在解决了吗

[chjwhere]

你好，请问你解决了吗

[justinzhang528]

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

[chjwhere]

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