current_device

dragon.cuda.current_device()[source]

Return the index of current selected device.

Returns:
int – The device index.