BalancedDataParallel 如何指定显卡 about balanced-dataparallel HOT 6 CLOSED

用CUDA_VISIBLE_DEVICES关键字设置你要用的显卡

from balanced-dataparallel.

CUDA_VISIBLE_DEVICES

用了，但还是使用4块卡。

from balanced-dataparallel.

代码里面使用os.environ['CUDA_VISIBLE_DEVICES']进行设置你要使用的显卡

from balanced-dataparallel.

代码里面使用os.environ['CUDA_VISIBLE_DEVICES']进行设置你要使用的显卡

对 我就是这么设置的。 但还是用了4块卡。
部分代码如下：
os.environ['CUDA_VISIBLE_DEVICES'] = "0,2"

netD.to(gpu_ids[0])
netD = BalancedDataParallel(16,netD, dim=0)

在class BalancedDataParallel(DataParallel)中修改如下：
device_ids = [0,2]
inputs, kwargs = self.scatter(inputs, kwargs, device_ids)

from balanced-dataparallel.

那就不知道了，建议谷歌找答案

from balanced-dataparallel.

那就不知道了，建议谷歌找答案

你好，我通过'CUDA_VISIBLE_DEVICES=0,2 python ....' 解决了此问题。
目前设置0号卡batchsize为8 总batchsize为32. 但训练时间比之前直接使用torch.nn.DataParallel (batchsize 24)慢25%。
您知道如何解决吗？

from balanced-dataparallel.