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 时,将自动导入树外后端扩展。 |