Skip to content

Get Device

Source

Convenience functions using torch.

These functions are used to control certain aspects and behaviours of PyTorch.

get_device()

Get the device on which operations take place.

Returns:

  • str –

    The device on which operations take place, e.g. "cuda", "cpu" or "mps".