YOLOv3-complete-pruning icon indicating copy to clipboard operation
YOLOv3-complete-pruning copied to clipboard

Published 20 hours ago • verified publisher icon coldlarry

win10多显卡

Open AadeIT opened this issue 5 years ago • 4 comments

我在win10使用2张显卡，但是代码中只显示单显卡，我应该怎么做？pytorch-yolo的作者说在linux上可以使用，请问win怎么支持多显卡

AadeIT avatar Mar 11 '20 09:03 AadeIT

pytorch在win10不支持多显卡，我在网上看人说不支持，而且运行的话，它分布式训练那里会报错的。

RosabanksiaeAit avatar Mar 20 '20 13:03 RosabanksiaeAit

@RosabanksiaeAit 我在win10上面的显卡可以跑通只需要把cuda_count改变就可以，但是无法把多块GPU沾满

AadeIT avatar Mar 20 '20 13:03 AadeIT

我当时这里报错欸： if torch.cuda.device_count() > 1: dist.init_process_group 说是没有init_process_group，而且我在外国一个网站上看到有人说，还不支持windows。 要占满的话好像在代码里可以 添加很多改动，我在博客里看到过但是我看不太懂。就搜分布式训练啥的？CSDN上就很多博客。

RosabanksiaeAit avatar Mar 20 '20 13:03 RosabanksiaeAit

@RosabanksiaeAit 你把这里改成99就可以进行训练了，你尝试一下，但是无法把多块GPU吃满，还是一块在跑

AadeIT avatar Mar 20 '20 13:03 AadeIT