评价此页

torch.cuda.init#

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

初始化 PyTorch 的 CUDA 状态。

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

如果 CUDA 状态已初始化,则什么也不做。