megengine.get_device_count

get_device_count(device_type)

获取当前系统中的设备数量。

参数

device_type (str) – 设备类型，’gpu’ 和 ‘cpu’ 中的一个

返回类型

int

megengine.is_cuda_available megengine.get_default_device