其他环境变量

变量	说明
TORCH_FORCE_WEIGHTS_ONLY_LOAD	如果设置为 [1, y, yes, true]，torch.load 将使用 weight_only=True。有关此内容的更多文档，请参阅 torch.load()。
TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT	在某些条件下，自动梯度线程可能会在关闭时挂起，因此我们不会无限期地等待它们关闭，而是依赖于超时，超时默认设置为 10 秒。此环境变量可用于以秒为单位设置超时。