捷徑

其他環境變數

變數

描述

TORCH_FORCE_WEIGHTS_ONLY_LOAD

如果設置為 [1, y, yes, true],則 torch.load 將使用 weight_only=True。有關此內容的更多文件,請參閱 torch.load()

TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT

在某些情況下,自動微分執行緒可能會在關閉時掛起,因此我們不會無限期地等待它們關閉,而是依賴預設設置為 10 秒的逾時。此環境變數可用於設置以秒為單位的逾時。

文件

訪問 PyTorch 的全面開發者文件

查看文件

教學

獲取針對初學者和高級開發人員的深入教學

查看教學

資源

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

查看資源