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	等待給定裝置上所有流中的所有核心完成。