torch.cuda.init

torch.cuda.init()

初始化 PyTorch 的 CUDA 狀態。

如果您透過其 C API 與 PyTorch 互動，您可能需要明確呼叫此函數，因為 CUDA 功能的 Python 綁定在完成此初始化之前將無法使用。 普通用戶應該不需要這個，因為 PyTorch 的所有 CUDA 方法都會自動按需初始化 CUDA 狀態。

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