[Python] 현재 코드가 사용하는 cpu 확인

---

Pytorch를 이용해서 학습 등을 진행할 때 GPU 사용량은 보통 nvidia-smi로 확인한다.

반면 cpu를 확인하기 위해 htop등을 사용하기에는 불편함이 있다.

그럴 때 이렇게 하면 된다.

import os
import psutil

pid = os.getpid()
py  = psutil.Process(pid)

cpu_usage   = os.popen("ps aux | grep " + str(pid) + " | grep -v grep | awk '{print $3}'").read()
cpu_usage   = cpu_usage.replace("\n","")

memory_usage  = round(py.memory_info()[0] /2.**30, 2)

print("cpu usage\t\t:", cpu_usage, "%")
print("memory usage\t\t:", memory_usage, "%")