I did model=nn.DataParallel(model, device_ids=gpus).cuda(device_id=gpus[0])
All the data I pass to model exist on device 0 (gpus[0]). I still get arguments are located on different GPUS.
What’s wrong??