set_default_device#
- ivy.set_default_device(device, /)[source]#
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')