torch.accelerator

此套件引入對目前 加速器 在 python 中的支援。

device_count	傳回目前可用的 加速器 數量。
is_available	檢查是否有可用的 加速器。
current_accelerator	傳回目前 加速器 的裝置。
set_device_index	將目前裝置索引設定為給定的裝置。
set_device_idx	將目前裝置索引設定為給定的裝置。
current_device_index	傳回目前 加速器 所選裝置的索引。
current_device_idx	傳回目前 加速器 所選裝置的索引。
set_stream	將目前的 stream 設定為給定的 stream。
current_stream	傳回給定裝置目前選取的 stream。
synchronize	等待指定裝置上所有串流中的所有核心完成。