目录

快捷方式

torch.mtia.record_memory_history¶

torch.mtia.record_memory_history(enabled='all', stacks='python', max_entries=0)[源码][源码]¶

启用/禁用 MTIA 分配器上的内存分析器

参数

enabled (all 或 state, 可选) – 如果 device 为 None (默认)，则为 current_device() 给出的当前设备的统计信息。
stacks ("python" 或 "cpp", 可选) –
max_entries (int, 可选) –

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源