Shortcuts

Miscellaneous Environment Variables

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources