diff --git a/experiment_impact_tracker/gpu/nvidia.py b/experiment_impact_tracker/gpu/nvidia.py index ff8a04d..3dd5292 100644 --- a/experiment_impact_tracker/gpu/nvidia.py +++ b/experiment_impact_tracker/gpu/nvidia.py @@ -189,7 +189,7 @@ def get_nvidia_gpu_power(pid_list, logger=None, **kwargs): gpu_data["utilization"] = {"gpu_util": gpu_util, "memory_util": memory_util} # get power - power_readings = gpu.findall("power_readings")[0] + power_readings = gpu.findall("power_readings")[0] if gpu.findall("power_readings") else gpu.findall("gpu_power_readings")[0] power_draw = power_readings.findall("power_draw")[0].text gpu_data["power_readings"] = {"power_draw": power_draw}