Rate this Page

torch.cuda.current_device#

torch.cuda.current_device()[source]#

Return the index of a currently selected device.

Return type

int