CUDA out of memory. Tried to allocate 392.00 MiB (GPU 0； 10.76 GiB total capacity； 652.77 MiB alread

Posted 2021-09-04 刘润森！

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了CUDA out of memory. Tried to allocate 392.00 MiB (GPU 0； 10.76 GiB total capacity； 652.77 MiB alread相关的知识，希望对你有一定的参考价值。

在使用Pytorch中，出现CUDA out of memory. Tried to allocate 392.00 MiB (GPU 0; 10.76 GiB total capacity; 652.77 MiB alread

这是内存不够用的结果，将num_workers和batch size设置小点

以上是关于CUDA out of memory. Tried to allocate 392.00 MiB (GPU 0； 10.76 GiB total capacity； 652.77 MiB alread的主要内容，如果未能解决你的问题，请参考以下文章