https://zdevito.github.io/2022/08/16/memory-snapshots.html
# enable the recording of stack frame information for each allocation
import torch
torch.cuda.memory._record_memory_history(True)
from torchvision.models import resnet18
from pprint import pprint