1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
.. _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``.
|
