merlin.models.loader.utils.device_mem_size

merlin.models.loader.utils.device_mem_size(kind='total', cpu=False)