快捷方式

杂项环境变量

变量

描述

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],则 torch.load 将使用 weights_only=False(如果在调用点未传递 weights_only 变量)。有关更多文档,请参见 torch.load()

TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT

在某些条件下,autograd 线程在关闭时可能会挂起,因此我们不会无限期地等待它们关闭,而是依赖于默认设置为 10 秒的超时。此环境变量可用于设置超时时间(以秒为单位)。

TORCH_DEVICE_BACKEND_AUTOLOAD

如果设置为 1,则在运行 import torch 时,树外后端扩展将自动导入。

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源