快捷方式

雜項環境變數

變數

描述

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 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源