mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
e24871eb3c
In preparation for `weights_only` flip, if users don't have access to the `torch.load` call Pull Request resolved: https://github.com/pytorch/pytorch/pull/138225 Approved by: https://github.com/albanD
20 lines
1.2 KiB
ReStructuredText
20 lines
1.2 KiB
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 ``weights_only=True``. This will happen even if
|
|
``weights_only=False`` was passed at the callsite. For more documentation on this, see :func:`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 :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``.
|