雜項環境變數

創建於：2025年6月17日 | 最後更新於：2025年6月17日

變數	描述
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 時，將自動匯入樹外（out-of-tree）的後端擴充套件。