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	等待给定设备上所有流中的所有内核完成。