torch.cuda.set\_device
======================

.. currentmodule:: torch.cuda

.. autofunction:: set_device