torch.cuda.current_device

torch.cuda.current_device() [source]

Return the index of a currently selected device.

Return type

int