tomodrgn.utils.get_default_device

get_default_device() → device

Determine whether a CUDA-capable GPU is availble or not for torch operations.

Returns:

torch.device(‘cuda’) if CUDA-capable GPU is available, else torch.device(‘cpu’).