https://travelebookdownload.blogspot.com/2023/11/cuda-and-pytorch-memory-usage.html