I am reading the documentation of nn.DataParallel:


device_ids  ( list of python:int  or   torch.device  1) – CUDA devices (default: all devices)

And I can only set single device with code like  device = torch.device("cuda:0").
How to set multi-device to use the default: all devices?