PyTorch Forums
Difference between Cuda:0 vs Cuda with 1 GPU
PiedPiper
August 17, 2020, 2:41pm
6
Yes and if you want using multi GPUs you shall check with model.DataParallel() and model.to(device).
show post in topic