gnn_tracking.utils.device

gnn_tracking.utils.device#

Functions#

guess_device([inpt])

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

Module Contents#

gnn_tracking.utils.device.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

Contents