torch 加速器 ¶

本包引入了对当前 Python 加速器的支持。

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