評價此頁

torch.cuda.current_device#

torch.cuda.current_device()[原始碼]#

返回當前選定裝置的索引。

返回型別

int