psutil 사용해서 특정 cpu에 python 스크립트 할당하기

어떻게 특정 cpu를 할당할 수 있을까?

>>> import psutil
>>> psutil.cpu_count()
4
>>> p = psutil.Process()
>>> p.cpu_affinity()  # get
[0, 1, 2, 3]
>>> p.cpu_affinity([0])  # set; from now on, process will run on CPU #0 only
>>> p.cpu_affinity()
[0]
>>>
>>> # reset affinity against all CPUs
>>> all_cpus = list(range(psutil.cpu_count()))
>>> p.cpu_affinity(all_cpus)
>>>

• https://stackoverflow.com/questions/14716659/taskset-python/40856471#40856471

실험을 돌리던중 cpu 할당을 특정할 필요가 있어 찾아보던 중 psutil 라이브러리를 사용하면 가능하다는 것을 알았다. 위의 스크립트는 그 예시. 

 

즉 스크립트에 이 3 줄만 추가하면 원하는 cpu 코어에서만 실행하도록 할 수 있다!

import psutil
 
p = psutil.Process()
p.cpu_affinity([0])  # set; from now on, process will run on CPU #0 only
 

 

 

 

References

• https://stackoverflow.com/questions/14716659/taskset-python/40856471#40856471