import multiprocessing
from multiprocessing import Pool
# 导入 Pool 允许你创建一个进程池
# 进程池是一组工作进程,它们可以并行地执行多个任务
# multiprocessing.cpu_count(): 返回当前机器上的CPU核心数
sum_cpu = multiprocessing.cpu_count()
use_cpu = max(1, int(multiprocessing.cpu_count() * 0.5))
# 计划使用的CPU核心数
# 定义要并行执行的函数
def square(number):
return number * number
# 要处理的数字列表
numbers = [1, 2, 3, 4, 5]
# 创建进程池,并使用imap方法并行执行square函数
if __name__ == '__main__':
with Pool(processes=4) as pool: # 假设我们使用4个进程
squared_numbers = pool.imap(square, numbers)
# 列表numbers中的每一个元素,作为并行执行函数的参数输入
# 打印每个数字的平方
for squared_number in squared_numbers:
print(squared_number)
查看cpu进程数
于 2024-05-24 20:11:37 首次发布