oneflow.cuda.current_device

oneflow.cuda.current_device()int

Returns local rank as device index.