训练模型报错 Out of memory
收藏
按照官网给的例子，训练模型
python tools/train.py -c configs/yolov3/yolov3_mobilenet_v1_roadsign.yml --eval -o use_gpu=true
报错：
Out of memory error on GPU 0. Cannot allocate 18.062744MB memory on GPU 0, available memory is only 9.054688MB.
我显卡是 2G的 ，怎么可能会不足 18M ，不知道是啥问题，大佬们帮忙看看！谢谢
Medlen
已解决
3#
回复于2021-06
这个报错信息的意思是：你的显卡已经分配给程序一部分显存了，然后还想要再分配18M，但是由于之前占用显存太多，导致剩余显存不够，所以报错。这个你的2G显卡太小了，不建议用自己的电脑跑，感觉大部分程序都跑不了。如果是模型结构比较简单，并且没有使用预训练模型的话，你可以试着降低下batch_size试试。然后程序的话，建议使用AI studio，可以白嫖tesla显卡，由16G和32G显存的两种，这个多香呀。
0
收藏
请登录后评论
Out of memory
这个报错信息的意思是：你的显卡已经分配给程序一部分显存了，然后还想要再分配18M，但是由于之前占用显存太多，导致剩余显存不够，所以报错。这个你的2G显卡太小了，不建议用自己的电脑跑，感觉大部分程序都跑不了。如果是模型结构比较简单，并且没有使用预训练模型的话，你可以试着降低下batch_size试试。然后程序的话，建议使用AI studio，可以白嫖tesla显卡，由16G和32G显存的两种，这个多香呀。
妙啊
显存爆了，可以在飞桨上跑
batchsize调到1先，跑通再说
报错是说再分配那么多内存时，显存不够了，不是说一共那么多显存