Pytorchにはtorch.deviceとDataParallel療法で使用するGPUを指定可能ですが、それらの設定が矛盾している場合はどうなるのでしょうか?
例:
gpus = (0,2,4) #使用するGPU番号のtuple(複数選択可)
⇓ GPU id を0,2,4に設定された!
device = torch.device(f"cuda:{min(gpus)}" if len(gpus) > 0 else 'cpu')
⇓ GPU を1,3,5に指定された!
model = DataParallel(model, device_ids=[1,3,5])
model.to(device)
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。