RuntimeError: CUDA out of memory

RuntimeError: CUDA out of memory

在进行深度学习任务时,我们可能会遇到这个错误以及相关的内存提示。
在我查看了大量比较常规的解决方案后,发现并不能解决问题(当然,以我电脑的配置是不会因为数据量的问题报错),所以笔者在这里提供另一种解决方式。

情况介绍:在代码刚开始跑的时候(甚至是调试的时候)就会报错内存相关的,如下

RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 2.00 GiB total capacity; 1.29 GiB already allocated; 79.00 MiB free; 1.30 GiB reserved in total by PyTorch)

笔者的解决方案:
1.打开系统属性
在这里插入图片描述
2.选择高级,并修改驱动器虚拟内存,一般设置为100000MB就足够了(PS:笔者的pycharm以及代码都是放在D盘,所以这里只修改了D,如果存放位置不同,建议调整其他盘虚拟内存)
在这里插入图片描述
3.点击设置并应用
在这里插入图片描述

posted @ 2023-03-17 12:21  yokon  阅读(276)  评论(0)    收藏  举报  来源