mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
312652c325
Fixes #122468 - Load device extensions at the end of `torch/__init__.py` - Enabled by default, or you can disable it with `TORCH_DEVICE_BACKEND_AUTOLOAD=0` run test: ```python python test/run_test.py -i test_autoload_enable python test/run_test.py -i test_autoload_disable ``` doc: https://docs-preview.pytorch.org/pytorch/pytorch/127074/miscellaneous_environment_variables.html co-author: @jgong5 @bsochack @bkowalskiINTEL @jczaja @FFFrog @hipudding Co-authored-by: albanD <desmaison.alban@gmail.com> Co-authored-by: Jiong Gong <jiong.gong@intel.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/127074 Approved by: https://github.com/albanD, https://github.com/jgong5
16 lines
831 B
ReStructuredText
16 lines
831 B
ReStructuredText
.. _miscellaneous_environment_variables:
|
|
|
|
Miscellaneous Environment Variables
|
|
===================================
|
|
.. list-table::
|
|
:header-rows: 1
|
|
|
|
* - Variable
|
|
- Description
|
|
* - ``TORCH_FORCE_WEIGHTS_ONLY_LOAD``
|
|
- If set to [``1``, ``y``, ``yes``, ``true``], the torch.load will use ``weight_only=True``. For more documentation on this, see :func:`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 timeout that is 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``.
|