已解决:Windows系统运行yolov10训练时报错:freeze_support()以及GPU_mem=0

已解决:Windows系统运行yolov10训练时报错:freeze_support()以及GPU_mem=0@TOC

之前用实验室的服务器跑v10的时候按照github上的四个步骤安装环境,很顺利的就跑了起来,但是后来在我自己的电脑上使用同样的步骤再次训练时遇到了下面的问题,虽然能够正常运行,但是调用不了多线程,速度很慢,查看后台发现GPU根本没动,终端也是显示GPU_mem为0,为此上网找了好久的资料,发现有人说是pytorch和cuda版本没对应,或是说用了外置源的问题,为此我配环境无数次花了几个小时也没解决问题。在这里插入图片描述后来又看到一种说法是先把requirements.txt中的前两项注释掉,手动安装pytorch,但接下来就报下面的错误:在这里插入图片描述为此我又开始了寻找解决的办法,终于皇天不负有心人,还真让我找到了问题所在,原来是因为yolov10的代码对Windows不友好,语法不太一样,只要在训练文件中添加一行代码

`if __name__ == '__main__':

即可正常调用GPU训练,但此过程中我发现运存暴涨了一会后,GPU才开始正常工作,不知其因,但终归可以正常训练,如下,希望对你有所帮助。
图片: 在这里插入图片描述

  • 25
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值