Source code for ts.metrics.process_memory_metric

Collect process memory usage metrics  here
Pass a json, collection of pids and gpuID

import logging

import psutil

[docs]def get_cpu_usage(pid): """ use psutil for cpu memory :param pid: str :return: int """ try: process = psutil.Process(int(pid)) except psutil.Error: logging.error("Failed get process for pid: %s", pid, exc_info=True) return 0 mem_utilization = process.memory_info()[0] return mem_utilization
[docs]def check_process_mem_usage(stdin): """ Return ------ mem_utilization: float """ process_list = stdin.readline().strip().split(",") for process in process_list: if not process: continue"%s:%d", process, get_cpu_usage(process))


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources