https://jdhao.github.io/2018/04/02/pytorch-gpu-usage/