torch.cuda.set_sync_debug_mode

torch.cuda.set_sync_debug_mode(debug_mode) [source]

Устанавливает режим отладки для операций синхронизации cuda.

Параметры:

debug_mode (str или int) – если “default” или 0, не выводить ошибки или предупреждения при операциях синхронизации, если “warn” или 1, выводить предупреждения при операциях синхронизации, если “error” или 2, выводить ошибки при операциях синхронизации.

Предупреждение

Это экспериментальная функция, и не все операции синхронизации будут вызывать предупреждения или ошибки. В частности, операции в пространствах имен torch.distributed и torch.sparse пока не поддерживаются.