torchprof - 用于Pytorch模型的逐层分析的最小依赖库-python

torchprof - 用于Pytorch模型的逐层分析的最小依赖库 torchprof 用于 Pytorch 模型的逐层分析的最小依赖库。 所有指标都是使用 PyTorch autograd 分析器得出的。 快速入门 pip install torchprof import torch import torchvision import torchprof model = torchvision.models.alexnet(pretrained=False).cuda() x = torch.rand([1, 3, 224, 224]).cuda() with torchprof.Profile (model, use_cuda=True) as prof: model(x) print(prof.display(show_events=False)) # 等价于 `print(prof)` 和 `print(prof.display())` 模块 | 自 CPU 总量 | CPU总量| CUDA 总数 | 发生 ---------------|------------