近日，PyTorch 核心开发者和 FAIR 研究者 Zachary DeVito 创建了一个新工具（添加实验性 API），通过生成和可视化内存快照（memory snapshot）来可视化 GPU 内存的分配状态。这些内存快照记录了内存分配的堆栈跟踪以及内存在缓存分配器状态中的位置。
接下来，通过将这些内存快照可视化为火焰图（flamegraphs），内存的使用位置也就能一目了然地看到了。
图灵奖得主 Yann Lecun 也转推了这个工具。
Twitter@Zachary DeVito
工具链接：https://zdevito.github.io/2022/08/16/memory-snapshots.html
内容中包含的图片若涉及版权问题，请及时与我们联系删除
评论
沙发等你来抢