torch.cuda.init

torch.cuda.init()

初始化 PyTorch 的 CUDA 狀態。

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

如果 CUDA 狀態已初始化，則不執行任何操作。