torch.cuda.init

torch.cuda.init()

初始化 PyTorch 的 CUDA 状态。

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

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