python指定cpu使用率，與內存占用率

 

python指定cpu使用率，與內存占用率

"""
runing.py -c 2 -t 0.01 -m 1000
-c cpu核數，不加-c參數為最大核數
-t cpu運算頻率時間，間隔，越小占用越高
-m 內存占用，1000MB
"""
import sys
import time
from time import clock
import argparse
from multiprocessing import Process
from multiprocessing import cpu_count
import math
# import tempfile

# # 屏蔽 Traceback信息
# sys.stderr=tempfile.TemporaryFile() 


def exec_func(bt):

    while True: 
        for i in range(0, 9600000):
            pass
        time.sleep(bt)



if __name__ == "__main__":

    parse = argparse.ArgumentParser(description='runing')

    parse.add_argument(
        "-c",
        "--count",
        default= cpu_count(),
        help='cpu count'
        )

    parse.add_argument(
        "-t",
        "--time",
        default= 0.01,
        help='cpu time'
        )
    parse.add_argument(
        "-m",
        "--memory",
        default= 1000,
        help='memory'
        )


    args = parse.parse_args()

    cpu_logical_count = int(args.count)

    cpu_sleep_time = args.time

    memory_used_mb = int(args.memory)

    try:
        cpu_sleep_time = int(args.time)
    except ValueError:
        try:
            cpu_sleep_time = float(args.time)
        except ValueError as ex:
            raise ValueError(ex)

    _doc = """
            runing.py -c 2 -t 0.01 -m 1000
            -c 指定cpu核數，不加-c參數默認為當前cpu最大核數
            -t cpu運算頻率時間，間隔，越小占用越高
            -m 內存占用，1000MB

            CPU使用率需要手動增加減少-t參數來達到，預期使用率。
           """
    print("\n====================使用說明=========================")
    print("{}".format(_doc))
    print("\n====================================================")
    print('\n當前占用CPU核數:{}'.format(cpu_logical_count))
    print('\n內存預計占用:{}MB'.format(memory_used_mb))
    print('\n資源浪費中......')

    try:
        # 內存占用
        s = ' ' * (memory_used_mb * 1024 * 1024)
    except MemoryError:
        print("剩余內存不足，內存有溢出......")

    try:

        p = Process(target=exec_func, args=("bt",))

        ps_list = []

        for i in range(0, cpu_logical_count):
            ps_list.append(Process(target=exec_func, args=(cpu_sleep_time,)))

        for p in ps_list:
            p.start()

        for p in ps_list:
            p.join()

    except KeyboardInterrupt:
        print("資源浪費結束!")

 

新測可用：

 

 

 結果：

 

 

 

命令行模式，通過ctrl+c來結束