https://discountsnowblowersthisinstant.blogspot.com/2022/06/cuda-and-pytorch-memory-usage.html