When setting the backend to torch.cuda, it seems that only GPU0 can be used for computation. How to set up multiple Gpus for simultaneous computation
When setting the backend to torch.cuda, it seems that only GPU0 can be used for computation. How to set up multiple Gpus for simultaneous computation