跳到正文
原文
Hugging Face Blog·· 2026-05-29精选AI 评分62

PyTorch 性能分析(一):torch.profiler 入门指南

Profiling in PyTorch (Part 1): A Beginner's Guide to torch.profiler

AI 导读

Hugging Face 博客发布 PyTorch profiling 系列第一篇,以矩阵乘法加加法为例讲解 torch.profiler 的用法。文章说明 profiler 会导出统计表格和 Perfetto 时序 trace 两类产物,并演示如何用 warmup 消除冷启动开销、区分 CPU 与 GPU lane 以及读懂嵌套的调度链,最后附上速查表。

推荐理由

面向初学者的 torch.profiler 阅读指南,用两个矩阵算子串起从 CPU 调度到 CUDA 内核的完整链路,并附上速查表。

来源:Hugging Face Blog · huggingface.co