torch.cuda.init

torch.cuda.init()

初始化 PyTorch 的 CUDA 状态。

如果你通过 C API 与 PyTorch 交互，你可能需要显式调用此函数，因为在初始化完成之前，PyTorch 的 CUDA 功能的 Python 绑定将不可用。普通用户通常不需要此函数，因为 PyTorch 的所有 CUDA 方法都会在需要时自动初始化 CUDA 状态。

如果 CUDA 状态已初始化，则此函数不执行任何操作。