評價此頁

torch.cuda.init#

torch.cuda.init()[原始碼]#

初始化 PyTorch 的 CUDA 狀態。

如果您正在透過 C API 與 PyTorch 互動,可能需要顯式呼叫此函式,因為在執行此初始化之前,CUDA 功能的 Python 繫結將不可用。普通使用者通常不需要此函式,因為 PyTorch 的所有 CUDA 方法都會按需自動初始化 CUDA 狀態。

如果 CUDA 狀態已初始化,則什麼也不做。