快捷方式

torch.cuda.init

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

初始化 PyTorch 的 CUDA 状态。

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

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答问题

查看资源