快捷方式

torch.accelerator

此包引入了对当前 加速器 在 Python 中的支持。

device_count

返回当前可用的 加速器 数量。

is_available

检查当前加速器在运行时是否可用:它是否已构建、所有必需的驱动程序是否都已安装,以及至少有一个设备是否可见。

current_accelerator

返回编译时可用的加速器设备。

set_device_index

将当前设备索引设置为给定的设备。

set_device_idx

将当前设备索引设置为给定的设备。

current_device_index

返回当前 加速器 当前选定设备的索引。

current_device_idx

返回当前 加速器 当前选定设备的索引。

set_stream

将当前流设置为给定的流。

current_stream

返回给定设备当前选定的流。

synchronize

等待给定设备上所有流中的所有内核完成。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源