杂项环境变量 ¶

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