chainer.get_device¶
- chainer.get_device(device_spec: Union[backend.Device, chainerx.Device, cuda.Device, str, Tuple[str, int], ModuleType, Tuple[ModuleType, int]]) chainer._backend.Device [source]¶
Returns a device object.
- Parameters
device_spec (object) –
Device specifier. If a
chainer.backend.Device
instance is given, it is returned intact. Otherwise the following values are supported:ChainerX devices
A string representing a device. (ex.
'native:0'
,'native'
)A
chainerx.Device
object.
CuPy
A string starts with
'@cupy:'
. (ex.'@cupy:0'
)A
cupy.cuda.Device
object.
NumPy
The string
'@numpy'
.
NumPy with Intel Architecture
The string
'@intel64'
.