快捷方式

torch.accelerator.current_device_idx

torch.accelerator.current_device_idx()[原始碼]

傳回目前 加速器 所選裝置的索引。

返回值

目前所選裝置的索引。

回傳類型

int

文件

存取 PyTorch 的完整開發人員文件

檢視文件

教學

取得適用於初學者和進階開發人員的深入教學

檢視教學

資源

尋找開發資源並取得問題解答

檢視資源