YOLO V5怎么多GPU同时训练一个项目?可以解决显存不够的问题吗?
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
default里面写上需要调用的gpu就行了,0代表装在主板上接口0的显卡,0,1代表调用第一张和第二张,以此类推。