distrax.devices.set_device

distrax.devices.set_device(device)

Sets the DEVICE to use.

Parameters:

device (str) – the device to get, i.e. brd

Return type:

None

Examples

>>> distrax.devices as device
>>> device.set_device()