torch.accelerator

创建于：2024 年 10 月 27 日 | 最后更新于：2025 年 7 月 10 日

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

device_count	返回当前可用加速器的数量。
is_available	检查当前加速器在运行时是否可用：它是否已构建，所有必需的驱动程序是否可用以及至少一个设备是否可见。
current_accelerator	返回编译时可用加速器的设备。
set_device_index	将当前设备索引设置为给定设备。
set_device_idx	将当前设备索引设置为给定设备。
current_device_index	返回当前加速器的当前选定设备的索引。
current_device_idx	返回当前加速器的当前选定设备的索引。
set_stream	将当前流设置为给定流。
current_stream	返回给定设备的当前选定流。
synchronize	等待给定设备上所有流中的所有内核完成。
device_index	用于设置当前加速器的当前设备索引的上下文管理器。