megengine.get_device_count

get_device_count(device_type)

Gets number of devices installed on this system.

Parameters

device_type (str) – device type, one of ‘gpu’ or ‘cpu’

Return type

int

previous

megengine.is_rocm_available

next

megengine.get_default_device