YOLO V5怎么多GPU同时训练一个项目？

Question

YOLO V5怎么多GPU同时训练一个项目？可以解决显存不够的问题吗？

Answer 1

    parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')

default里面写上需要调用的gpu就行了，0代表装在主板上接口0的显卡，0,1代表调用第一张和第二张，以此类推。