CUDA out of memory
Opened this issue · 2 comments
amiaynara commented
Tried to allocate 20.00 MiB (GPU 0; 1.96 GiB total capacity; 1.24 GiB already allocated; 2.88 MiB free; 1.25 GiB reserved in total by PyTorch)
zijunsun commented
maybe you can use batch accumulate to decrease the batch size.
huanghao-att commented
Tried to allocate 20.00 MiB (GPU 0; 1.96 GiB total capacity; 1.24 GiB already allocated; 2.88 MiB free; 1.25 GiB reserved in total by PyTorch)
你是在window上跑的嘛