PyTorch only using one GPU?
From nvidia-smi, I can see that during training, my pyTorch script is only using one GPU. Is there a way to make it use the others? (I have two). Thanks.
Kalamaya
apaszke
Now in my main script, I call: I took a look at the documentation for the DataParallel, and what I currently do is this: Now this seems to run, BUT, it complains that it doesnt know what “forward” is. (The forward member function that it). So I am confused as to what exactly I should be putting through this “DataParallel” exactly… every member function of my DNN?.. Thanks
Kalamaya
apaszke