torch.cuda.init

torch.cuda.init()[源代码]

初始化 PyTorch 的 CUDA 状态。

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

如果 CUDA 已经初始化，则不做任何操作。