What torch.cuda.set_device() does?
If it sets the GPU for execution, how can I set multiple GPUs to run my experiment? For example, I want to tell to pytorch that you should use two GPUs (if available) to run my experiment. How can I achieve that?
wasiahmad
xwgeng
Can you give an example in order to use multi-GPU?
wasiahmad
xwgeng