这个屌丝很懒，什么也没留下！

pytorch：out of Memory 问题以及pytorch训练时卡死问题_pytorch executioncontext.cpp::executioncontext::21

作者：盐析白兔 | 2024-03-11 20:51:55

pytorch executioncontext.cpp::executioncontext::213] error code 2: outofmemo

1 ）pytorch：out of Memory 问题

“RuntimeError: CUDA out of memory. Tried to allocate 10.54 GiB (GPU 0;
31.72…”

此问题在我调小了batch_size后得到解决

2 ）pytorch训练时卡死问题
参考链接
此问题在我将dataloader中的work_threads调小后解决。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/220586