杂项环境变量

创建于：2025年6月17日 | 最后更新于：2025年6月17日

变量	描述
TORCH_FORCE_WEIGHTS_ONLY_LOAD	如果设置为 [1, y, yes, true]，torch.load 将使用 weights_only=True。即使在调用点传递了 weights_only=False，也会发生这种情况。有关更多信息，请参阅 torch.load。
TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD	如果设置为 [1, y, yes, true]，并且在调用点未传递 weights_only 变量，则 torch.load 将使用 weights_only=False。有关更多信息，请参阅 torch.load。
TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT	在某些情况下，autograd 线程可能在关闭时挂起，因此我们不会无限期地等待它们关闭，而是依赖一个默认设置为 10 秒的超时。此环境变量可用于设置超时（以秒为单位）。
TORCH_DEVICE_BACKEND_AUTOLOAD	如果设置为 1，当运行 import torch 时，将自动导入树外（out-of-tree）的后端扩展。