torch.cuda.set_sync_debug_mode

torch.cuda.set_sync_debug_mode(debug_mode)[source][source]

设置 cuda 同步操作的调试模式。

参数:

debug 模式（字符串或整数）- 如果为“默认”或 0，则在同步操作时不会报错或警告；如果为“警告”或 1，则在同步操作时警告；如果为“错误”或 2，则在同步操作时报错。

警告

这是一个实验性功能，并非所有同步操作都会触发警告或错误。特别是 torch.distributed 和 torch.sparse 命名空间中的操作尚未涵盖。