快捷方式

雜項環境變數

變數

描述

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 時,會自動匯入樹狀結構外的後端擴充。

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得適合初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並取得您的問題解答

檢視資源