快捷方式

torch._logging

PyTorch 具有可配置的日志记录系统,不同的组件可以设置不同的日志级别。例如,可以完全禁用一个组件的日志消息,而另一个组件的日志消息可以设置为最大详细程度。

警告

此功能处于 Beta 版,未来版本可能会有破坏兼容性的更改。

警告

此功能尚未扩展到控制 PyTorch 中所有组件的日志消息。

有两种方法可以配置日志记录系统:通过环境变量 TORCH_LOGS 或 Python API torch._logging.set_logs。

set_logs

为各个组件设置日志级别,并切换各个日志工件类型。

环境变量 TORCH_LOGS 是以逗号分隔的 [+-]<component> 对列表,其中 <component> 是下面指定的组件。+ 前缀将降低组件的日志级别,显示更多日志消息,而 - 前缀将提高组件的日志级别,显示更少的日志消息。默认设置是在 TORCH_LOGS 中未指定组件时的行为。除了组件之外,还有工件。工件是与组件关联的特定调试信息,可以显示或不显示,因此为工件添加前缀 +- 将不起作用。由于它们与组件关联,因此启用该组件通常也会启用该工件,除非该工件被指定为 off_by_default。此选项在 _registrations.py 中为那些过于冗余的工件指定,这些工件仅应在显式启用时显示。以下组件和工件可以通过 TORCH_LOGS 环境变量配置(有关 Python API,请参阅 torch._logging.set_logs)

组件
all

特殊组件,用于配置所有组件的默认日志级别。默认值:logging.WARN

dynamo

TorchDynamo 组件的日志级别。默认值:logging.WARN

aot

AOTAutograd 组件的日志级别。默认值:logging.WARN

inductor

TorchInductor 组件的日志级别。默认值:logging.WARN

your.custom.module

任意未注册模块的日志级别。提供完全限定名称,模块将被启用。默认值:logging.WARN

工件
bytecode

是否从 TorchDynamo 发出原始和生成的字节码。默认值:False

aot_graphs

是否发出 AOTAutograd 生成的图。默认值:False

aot_joint_graph

是否发出 AOTAutograd 生成的联合前向-后向图。默认值:False

compiled_autograd

是否从 compiled_autograd 发出日志。默认值:False

ddp_graphs

是否发出 DDPOptimizer 生成的图。默认值:False

graph

是否以表格格式发出 TorchDynamo 捕获的图。默认值:False

graph_code

是否发出 TorchDynamo 捕获的图的 Python 源代码。默认值:False

graph_breaks

是否在 TorchDynamo 跟踪期间遇到唯一图中断时发出消息。默认值:False

guards

是否为每个编译函数发出 TorchDynamo 生成的保护。默认值:False

recompiles

每次 TorchDynamo 重新编译函数时是否发出保护失败原因和消息。默认值:False

output_code

是否发出 TorchInductor 输出代码。默认值:False

schedule

是否发出 TorchInductor 调度。默认值:False

示例

TORCH_LOGS="+dynamo,aot" 将 TorchDynamo 的日志级别设置为 logging.DEBUG,并将 AOT 设置为 logging.INFO

TORCH_LOGS="-dynamo,+inductor" 将 TorchDynamo 的日志级别设置为 logging.ERROR,并将 TorchInductor 设置为 logging.DEBUG

TORCH_LOGS="aot_graphs" 将启用 aot_graphs 工件

TORCH_LOGS="+dynamo,schedule" 将 TorchDynamo 的日志级别设置为 logging.DEBUG,并启用 schedule 工件

TORCH_LOGS="+some.random.module,schedule" 将 some.random.module 的日志级别设置为 logging.DEBUG,并启用 schedule 工件

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源