快捷方式

torch.cuda.set_sync_debug_mode

torch.cuda.set_sync_debug_mode(debug_mode)[源代码]

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

参数

debug_mode (strint) – 如果为“default”或 0,则在同步操作时不报错也不警告;如果为“warn”或 1,则在同步操作时发出警告;如果为“error”或 2,则在同步操作时报错。

警告

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源