https://topminisite.com/blog/how-to-use-multiple-gpus-in-pytorch