MCPcopy Create free account
hub / github.com/mlco2/codecarbon / set_RAM_tracking

Method set_RAM_tracking

codecarbon/core/resource_tracker.py:24–40  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

22 self.tracker = tracker
23
24 def set_RAM_tracking(self):
25 logger.info("[setup] RAM Tracking...")
26 if self.tracker._force_ram_power is not None:
27 self.ram_tracker = (
28 f"User specified constant: {self.tracker._force_ram_power} Watts"
29 )
30 logger.info(
31 f"Using user-provided RAM power: {self.tracker._force_ram_power} Watts"
32 )
33 else:
34 self.ram_tracker = "RAM power estimation model"
35 ram = RAM(
36 tracking_mode=self.tracker._tracking_mode,
37 force_ram_power=self.tracker._force_ram_power,
38 )
39 self.tracker._conf["ram_total_size"] = ram.machine_memory_GB
40 self.tracker._hardware: List[Union[RAM, CPU, GPU, AppleSiliconChip]] = [ram]
41
42 def _setup_cpu_load_mode(self, tdp, max_power):
43 """Set up CPU tracking in load mode using psutil."""

Calls 1

RAMClass · 0.90

Tested by 1