torch.cuda.current\_device ========================== .. currentmodule:: torch.cuda .. autofunction:: current_device