chainerx.set_default_device

chainerx.set_default_device(device)

Sets the given device as the default device of the current thread.

Parameters

device (Device or str) – Device object or device name to set as the default device.

See also

• chainerx.get_default_device()

• chainerx.using_device()