mindspore.hal.device_count

View Source On Gitee
mindspore.hal.device_count(device_target=None)[source]

Returns device count of specified backend.

Note

If device_target is not specified, get the device count of the current backend set by context. For CPU backend, this method always returns 1.

Parameters

device_target (str, optional) – The device name of backend, should be one of “CPU”, “GPU” and “Ascend”.

Returns

int.

Examples

>>> import mindspore as ms
>>> device_target = ms.context.get_context("device_target")
>>> print(ms.hal.device_count(device_target))