So I can specify multiple GPUs in the .cuda() call? Because if I have to specify only one, which one should I specify?
In any case I usually use all the GPUs, and if I want to restrict pytorch access I just set CUDA_VISIBLE_DEVICES to the devices I want to use