torch.cuda.init

torch.cuda.init()

初始化 PyTorch 的 CUDA 状态。

如果您通过 PyTorch 的 C API 进行交互，可能需要显式调用此操作，因为 Python 绑定的 CUDA 功能将在此初始化完成后才可用。普通用户通常不需要这样做，因为 PyTorch 的所有 CUDA 方法都会在需要时自动初始化 CUDA 状态。

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