We usually use the below Python code to get CPU cores:
from multiprocessing import cpu_count print("CPU cores:", cpu_count())
But when the snippet running inside a docker container, it will just return the number of CPU cores for the physical machine the container runs on, not the actually
--cpus (for docker) or CPU limit (for Kubernetes).
Then, how could we get the CPU cores set by docker argument or Kubernetes configuration for this container?
The only answer I could find is here. And the corresponding Python code written by me is:
def get_cpu_limit(): with open("/sys/fs/cgroup/cpu/cpu.cfs_quota_us") as fp: cfs_quota_us = int(fp.read()) with open("/sys/fs/cgroup/cpu/cpu.cfs_period_us") as fp: cfs_period_us = int(fp.read()) container_cpus = cfs_quota_us // cfs_period_us # For physical machine, the `cfs_quota_us` could be '-1' cpus = cpu_count() if container_cpus < 1 else container_cpus return cpus