torch.cuda.set\_device ====================== .. currentmodule:: torch.cuda .. autofunction:: set_device