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