I’m trying to use the torch dataparallel class with specified GPU devices. When I set the torch.device as torch.device(“cuda:0,1”). Pytorch gives me the following error: RuntimeError: Invalid device string: ‘cuda:0,1’.
I have two GPU devices on my machine. Using Windows 10 OS.
I will look forward to any suggestions.