yolov5运行train.py出错

img


yolov5安装好后,运行train.py出现这种错误怎么解决啊

提示信息是 git 没有安装好,或者其环境变量没有配置好。

  • 你可以参考下这个问题的回答, 看看是否对你有帮助, 链接: https://ask.csdn.net/questions/7399264
  • 我还给你找了一篇非常好的博客,你可以看看是否有帮助,链接:yolov4_u5版复现—4. 训练 train.py
  • 除此之外, 这篇博客: yolov5 train.py中的 多机多卡  部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  • # DP mode
        if cuda and RANK == -1 and torch.cuda.device_count() > 1:
            logging.warning('DP not recommended, instead use torch.distributed.run for best DDP Multi-GPU results.\n'
                            'See Multi-GPU Tutorial at https://github.com/ultralytics/yolov5/issues/475 to get started.')
            model = torch.nn.DataParallel(model)
    • DP模式:单机多卡,但是这个情况也会出现一些问题,例如主卡爆掉,其他卡利用率上不去。当然,这是要修改底层和与硬件相关的问题,怎么优化还得靠人家框架官方。
    • DDP模式:多机多卡,能解决DP不均衡,主卡爆掉的问题,也就是单机多卡也能用,但是这个一般人也用不到,家里有条件的时候再去研究吧。
  • 您还可以看一下 白勇老师的YOLOv5(PyTorch)目标检测:原理与源码解析课程中的 train.py代码解析2小节, 巩固相关知识点