已解决: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才开始正常工作,不知其因,但终归可以正常训练,如下,希望对你有所帮助。
图片: