pytorch/docs/source/miscellaneous_environment_variables.rst

.. _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``.