小白求教 训练模型 刚开始正常 两三秒之后就报错RuntimeError: CUDA out of memory.

[wangxiang125488]

小白求教 训练模型 刚开始正常 两三秒之后就报错

• RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.02 GiB already allocated; 3.96 MiB free; 1.06 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

然后便自行终止 求大神帮忙看看 怎样解决

[HuaHuaOfficial]

显卡显存太小了。这个至少要4GB差不多

[HardToRegister]

我是4GB 的显存为何也报同样的错？

[beyondchenlin]

最低要求4g，有方法改，但建议提高显存更方便

Lafayette @.***> 于2022年12月11日周日 23:28写道：

我是4GB 的显存为何也报同样的错？

— Reply to this email directly, view it on GitHub https://github.com/babysor/MockingBird/issues/340#issuecomment-1345583201, or unsubscribe https://github.com/notifications/unsubscribe-auth/AZ6O4WWYLA2M4DPWDD2YKGTWMXXJ7ANCNFSM5L6EJW6A . You are receiving this because you are subscribed to this thread.Message ID: @.***>

-- jackchen