
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小节, 巩固相关知识点