yolov4-pytorch icon indicating copy to clipboard operation
yolov4-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

GPU训练问题

Open 3097254259 opened this issue 2 years ago • 5 comments

我开了cuda，但是在GTX1660和3090上训练速度差不多，这是怎么回事呢

3097254259 avatar May 08 '22 07:05 3097254259

具体的cuda和cudnn之类的配置？

bubbliiiing avatar May 08 '22 13:05 bubbliiiing

感谢回复，配置如下，四种都试过，速度都是一样的，速度都是与GTX1660相当，： image image image image

3097254259 avatar May 08 '22 14:05 3097254259

两种显卡 3090 和 A4000 训练速度基本没有提升

3097254259 avatar May 08 '22 14:05 3097254259

貌似是因为CPU性能过弱…你试试YoloV5看看

bubbliiiing avatar May 12 '22 14:05 bubbliiiing

我在网上尝试的时候CPU影响蛮大的

bubbliiiing avatar May 12 '22 14:05 bubbliiiing