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