Переменные среды для отладки

TORCH_SHOW_CPP_STACKTRACES

Если установлено значение 1, PyTorch выводит трассировку стека при обнаружении ошибки C++.

TORCH_CPP_LOG_LEVEL

Задает уровень журналирования средства c10 (поддерживаются как GLOG, так и средства журналирования c10). Допустимые значения: INFO, WARNING, ERROR и FATAL либо их числовые эквиваленты 0, 1, 2 и 3.

TORCH_LOGS

Подробнее об этой переменной среды см. в разделе torch._logging.

TORCH_DIST_NAN_CHECK

Если установлено значение 1, проверяет входные буферы (буферы отправки) каждой коллективной операции torch.distributed на наличие NaN для любого бэкенда. В отличие от TORCH_NCCL_NAN_CHECK, реализованного непосредственно в бэкенде NCCL, эта проверка выполняется с помощью не зависящих от бэкенда хуков коллективных операций ProcessGroup.