get\_device
===========

.. currentmodule:: ray.train.torch

.. autofunction:: get_device