pylops_gpu.utils.backend.device

pylops_gpu.utils.backend.device()[source]

Automatically identify device to be used with PyTorch

Returns:
device : str

Identified device, cpu or gpu