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

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代表调用第一张和第二张,以此类推。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 画两个图 python或R
  • ¥15 在线请求openmv与pixhawk 实现实时目标跟踪的具体通讯方法
  • ¥15 八路抢答器设计出现故障
  • ¥15 请教一下c语言的代码里有一个地方不懂
  • ¥15 opencv 无法读取视频
  • ¥15 用matlab 实现通信仿真
  • ¥15 按键修改电子时钟,C51单片机
  • ¥60 Java中实现如何实现张量类,并用于图像处理(不运用其他科学计算库和图像处理库))
  • ¥20 5037端口被adb自己占了
  • ¥15 python:excel数据写入多个对应word文档