mmengine.device

get_device	Returns the currently existing device type.
get_max_cuda_memory	Returns the maximum GPU memory occupied by tensors in megabytes (MB) for a given device.
is_cuda_available	Returns True if cuda devices exist.
is_npu_available	Returns True if Ascend PyTorch and npu devices exist.
is_mlu_available	Returns True if Cambricon PyTorch and mlu devices exist.
is_mps_available	Return True if mps devices exist.