https://dockerysedm310spring09.blogspot.com/2024/08/cuda-and-pytorch-memory-usage.html