Shortcuts

mmengine.device.get_device

mmengine.device.get_device()[源代码]

Returns the currently existing device type.

返回:

cuda | npu | mlu | mps | musa | cpu.

返回类型:

str