参考博客：
如何计算模型以及中间变量的显存占用大小：
https://oldpan.me/archives/how-to-calculate-gpu-memory
https://oldpan.me/archives/how-to-use-memory-pytorch
https://oldpan.me/archives/pytorch-gpu-memory-usage-track