it’s better you to use the environment variable CUDA_VISIBLE_DEVICES = id1,id2 and so on
it’s a cuda flag which will make any program to detect only those gpus you choose.
The input tensor should also be put on gpu devices, I can do this with input_tensor.to(device) if there is only single gpu device. How should I do that with multi-gpu?
The input tensor should also be put on gpu devices, I can do this with input_tensor.to(device) if there is only single gpu device. How should I do that with multi-gpu?