问题描述
最近在使用yolo跑目标检测的算法,3070的电脑用
python train.py --cfg yolov5n.yaml --batch-size 4 --epochs 1
跑batch4,就爆内存也太不合理了
解决方案
根据github issue作者的回答和这里的回答,可以判断是windows系统的问题
增加系统的虚拟内存就能解决,windows问题链接如下
解决步骤
以win11为例
根据windows官网描述:
-
在搜索框搜索控制面板
-
点系统和安全
-
点系统
-
点高级系统设置
5.点系统-性能-设置
6.点高级-虚拟内存-更改
7.取消【自动管理所有驱动器的分页大小】,下选【自定义大小】,我分配成了如下形式
8.分配完后一定要点【设置】按钮,之后重启电脑,可以在任务管理器-性能 看到【已提交】后备内存变多
9.之后再运行
python train.py --cfg yolov5n.yaml --batch-size 4 --epochs 1
就没问题啦