Дополнительные переменные среды

TORCH_FORCE_WEIGHTS_ONLY_LOAD

Если задано значение [1, y, yes, true], torch.load будет использовать weights_only=True. Это произойдёт, даже если weights_only=False было передано при вызове. Подробнее см. в документации: torch.load.

TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD

Если задано значение [1, y, yes, true], torch.load будет использовать weights_only=False, если переменная weights_only не была передана при вызове. Подробнее см. в документации: torch.load.

TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT

При некоторых условиях потоки autograd могут зависнуть при завершении работы, поэтому мы не ждём их завершения бесконечно, а используем тайм-аут, который по умолчанию равен 10 секундам. Эту переменную среды можно использовать для задания тайм-аута в секундах.

TORCH_DEVICE_BACKEND_AUTOLOAD

Если задано значение 1, расширения бэкенда, расположенные вне основного дерева исходного кода, будут автоматически импортироваться при запуске import torch.