喀喀喀吃 2024-04-06 18:16 采纳率: 66.7%
浏览 114
已结题

yolov8模型训练显示页面文件太小

cuda11.2，python3.8，安装好了pytorch，在pycharm上用yolov8训练，出现这个问题怎么办？求解。。

img

img


我在网上搜让分配虚拟内存，这样做了还是不行。是分配的太小了吗？那应该分配多少

img

  • 写回答

4条回答 默认 最新

  • 爱晚乏客游 2024-04-07 09:05
    关注

    cuda显存不足，将batch size设置为1或者2看下吧，两个G的显卡显存太小了，一般第一个epoch可能会需要更大的显存才行

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(3条)

报告相同问题？

问题事件

  • 系统已结题 4月15日
  • 已采纳回答 4月7日
  • 修改了问题 4月6日
  • 创建了问题 4月6日