当前位置:首页 > 服务端 > 解决CUDA out of memory

解决CUDA out of memory

问题

用Pytorch跑模型时,会出现RuntimeError: CUDA out of memory

解决方法

1.真的没有内存了

RuntimeError: CUDA out of memory. Tried to allocate 64.00 MiB 
(GPU 0; 4.00 GiB total capacity; 2.92 GiB already allocated; 58.76 MiB free; 
2.95 GiB reserved in total by PyTorch)

可以改小batch_size
2.内存足够,不分配

RuntimeError: CUDA out of memory. 
Tried to allocate 784.00 MiB 
(GPU 0; 4.00 GiB total capacity; 456.81 MiB already allocated; 2.47 GiB free; 
548.00 MiB reserved in total by PyTorch)

Win10下多线程的问题,num_threads/workers=0解决

作者:iClaire
来源链接:https://www.cnblogs.com/iclaire/p/12658730.html

版权声明:
1、Java侠(https://www.javaxia.com)以学习交流为目的,由作者投稿、网友推荐和小编整理收藏优秀的IT技术及相关内容,包括但不限于文字、图片、音频、视频、软件、程序等,其均来自互联网,本站不享有版权,版权归原作者所有。

2、本站提供的内容仅用于个人学习、研究或欣赏,以及其他非商业性或非盈利性用途,但同时应遵守著作权法及其他相关法律的规定,不得侵犯相关权利人及本网站的合法权利。
3、本网站内容原作者如不愿意在本网站刊登内容,请及时通知本站(javaclubcn@163.com),我们将第一时间核实后及时予以删除。





本文链接:https://www.javaxia.com/server/125789.html

标签: out of memory
分享给朋友: