torch.accelerator

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

device_count	返回当前可用的 加速器 数量。
is_available	检查是否有可用的加速器。
current_accelerator	返回当前加速器的设备。
set_device_idx	将当前设备索引设置为指定的设备。
current_device_idx	返回当前加速器下当前选定设备的索引。
set_stream	将当前流设置为指定的流。
current_stream	返回给定设备当前选定的流。
synchronize	等待给定设备上所有流中的所有内核完成。