set_default_device

ivy.set_default_device(device, /)

Set the default device to given device instance.

Parameters:

device (Union[Device, NativeDevice]) – The device to set as the default device

Return type:

None

Examples

>>> ivy.set_default_device("cpu")
>>> ivy.default_device()
'cpu'

>>> ivy.set_backend("torch")
>>> ivy.set_default_device("gpu:0")
>>> ivy.default_device(as_native=True)
device(type='cuda', index=0)

>>> import torch
>>> ivy.set_backend("torch")
>>> device = torch.device("cuda")
>>> ivy.set_default_device(device)
>>> ivy.default_device(as_native=True)
device(type='cuda')