其他环境变量¶ 变量 描述 TORCH_FORCE_WEIGHTS_ONLY_LOAD 如果设置为 [1, y, yes, true],则 torch.load 将使用 weight_only=True。有关此内容的更多文档,请参阅 torch.load()。 TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT 在某些情况下,autograd 线程可能会在关闭时挂起,因此我们不会无限期地等待它们关闭,而是依赖于默认设置为 10 秒的超时。此环境变量可用于设置以秒为单位的超时。 TORCH_DEVICE_BACKEND_AUTOLOAD 如果设置为 1,则在运行 import torch 时,将自动导入树外后端扩展。