frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
main
pytorch/docs/source/miscellaneous_environment_variables.md
windsonsea 7fcad0231c [Docs] Convert to markdown to fix 155025 (#155789) 
Related to #155025

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155789
Approved by: https://github.com/svekars
2025-06-17 15:08:14 +00:00

1.2 KiB
Raw Permalink Blame History

(miscellaneous_environment_variables)=

Miscellaneous Environment Variables

Variable Description
TORCH_FORCE_WEIGHTS_ONLY_LOAD If set to [1, y, yes, true], the torch.load will use weights_only=True. This will happen even if weights_only=False was passed at the callsite. For more documentation on this, see torch.load.
TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD If set to [1, y, yes, true], the torch.load will use weights_only=False if the weights_only variable was not passed at the callsite. For more documentation on this, see torch.load.
TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT Under some conditions, autograd threads can hang on shutdown, therefore we do not wait for them to shutdown indefinitely but rely on a timeout that is by default set to 10 seconds. This environment variable can be used to set the timeout in seconds.
TORCH_DEVICE_BACKEND_AUTOLOAD If set to 1, out-of-tree backend extensions will be automatically imported when running import torch.