z张大二
2021-05-11 10:28
采纳率: 100%
浏览 573

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

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

  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

1条回答 默认 最新

  • 爱晚乏客游 2021-05-11 11:05
    已采纳

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

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

    打赏 评论

相关推荐 更多相似问题