当前位置:   article > 正文

yolov5单机多卡训练加速配置_yolov5多卡

yolov5多卡

用默认单卡训练速度太慢,这里配置多卡,并使用缓存,内存得够用要不然无法缓存

--device 0,1 --cache

--device 0,1这里的0就是显卡1,1就是显卡2,如果你有更多显卡,可以继续往下写

训练过程

通过GPU-Z查看2个显卡的占用率

显卡1

显卡2

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/167775
推荐阅读
相关标签
  

闽ICP备14008679号