gnn_tracking.utils.device
=========================

.. py:module:: gnn_tracking.utils.device


Functions
---------

.. autoapisummary::

   gnn_tracking.utils.device.guess_device


Module Contents
---------------

.. py:function:: guess_device(inpt=None)

   Setting the device (cpu/cuda) with the following fallback alternatives

   1. input
   2. environment variable GNN_TRACKING_DEVICE
   3. cuda if available, else cpu