:py:mod:`gnn_tracking.utils.device`
===================================

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


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


Functions
~~~~~~~~~

.. autoapisummary::

   gnn_tracking.utils.device.guess_device



.. 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