首页
/ CloudSimPy:深度强化学习驱动的数据中心作业调度仿真框架

CloudSimPy:深度强化学习驱动的数据中心作业调度仿真框架

2024-09-16 00:44:16作者:廉皓灿Ida

项目介绍

CloudSimPy 是一个基于离散事件仿真框架 SimPy 的数据中心作业调度仿真框架,采用 Python 语言实现。Python 在科学计算、深度学习和机器学习领域拥有丰富的生态系统,这使得 CloudSimPy 能够无缝集成到 TensorFlow、PyTorch 等深度学习框架中,为研究基于机器学习或深度学习的资源管理方法提供了强大的支持。

项目技术分析

核心模块(Core)

core 包对数据中心作业调度问题中的各个实体进行了抽象和建模,包括任务实例、任务、作业、机器、计算集群、调度算法、调度器、代理、监控器等。这些模块通过策略模式的设计,使得调度算法的实现与调度器的实现相互独立,用户可以轻松自定义调度算法并应用于仿真中。

实验模块(Playground)

playground 包设计用于方便用户进行实验,支持考虑任务间依赖关系(DAG)和不考虑任务间依赖关系(Non_DAG)的仿真实验。预先实现了多种启发式作业调度算法及基于深度强化学习的作业调度算法,例如在 Non_DAG/algorithm/DeepJS 中实现的基于深度强化学习的数据中心作业调度算法。

高性能仿真

CloudSimPy 通过将 TaskInstance 设计为 SimPy 中的进程,实现了高效的仿真性能。状态信息的被动询问取代主动维护,优化了仿真过程中的 hotpath,使得仿真在关闭监测功能时能够快速高效地进行。

策略模式

CloudSimPy 广泛采用了策略模式,将调度算法的实现与调度器的实现独立开来。这种设计模式使得用户可以灵活地选择和切换不同的调度算法,而无需修改调度器的代码。此外,策略模式还被应用于奖励函数的实现,为不同的优化目标提供了不同的奖励计算方法。

项目及技术应用场景

CloudSimPy 适用于以下应用场景:

  1. 数据中心资源管理研究:通过仿真数据中心的作业调度过程,研究新的资源管理策略和算法。
  2. 深度强化学习算法验证:利用 CloudSimPy 进行基于深度强化学习的作业调度算法的研究和验证。
  3. 教学与培训:作为教学工具,帮助学生和研究人员理解数据中心作业调度的复杂性和挑战。

项目特点

  1. 强大的集成能力:无缝集成 Python 的科学计算、深度学习和机器学习生态,支持与 TensorFlow、PyTorch 等框架结合使用。
  2. 灵活的调度算法:通过策略模式,用户可以轻松自定义和切换不同的调度算法。
  3. 高效的仿真性能:通过被动询问状态信息,优化仿真过程中的性能,使得仿真在关闭监测功能时能够快速高效地进行。
  4. 丰富的实验支持:提供 playground 包,支持多种实验场景和预实现的调度算法,方便用户进行研究和实验。

结语

CloudSimPy 是一个功能强大且灵活的数据中心作业调度仿真框架,适用于各种研究和教学场景。无论你是研究人员、开发者还是学生,CloudSimPy 都能为你提供一个强大的工具,帮助你探索和优化数据中心的资源管理策略。赶快尝试一下吧!


参考文献

  1. DeepJS: Job Scheduling Based on Deep Reinforcement Learning in Cloud Data Center

运行示例

# 安装和运行
git clone git@github.com:RobertLexis/CloudSimPy.git
export PYTHONPATH=$PYTHONPATH:/path/to/cloudsimpy
cd cloudsimpy/playground/Non_DAG/launch_scripts
python main-makespan.py

依赖环境

  1. Python 3.6
  2. SimPy 3.0.11
  3. TensorFlow 1.12.0
  4. Numpy 1.15.3
  5. Pandas 0.23.4
登录后查看全文

项目优选

收起
kernelkernel
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
511
551
docsdocs
暂无描述
Markdown
852
5.69 K
ops-transformerops-transformer
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
1.05 K
2.49 K
kernelkernel
deepin linux kernel
C
33
16
pytorchpytorch
作为 Ascend for PyTorch 社区的核心组件,TorchNPU 是昇腾专为 PyTorch 打造的深度学习适配插件,使 PyTorch 框架能够直接调用昇腾 NPU,为开发者提供昇腾 AI 处理器的超强算力。
Python
839
1.28 K
ops-nnops-nn
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
847
1.7 K
ops-mathops-math
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.25 K
1.38 K
jiuwenswarmjiuwenswarm
JiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。
Python
3.17 K
857
AscendNPU-IRAscendNPU-IR
AscendNPU-IR是基于MLIR(Multi-Level Intermediate Representation)构建的,面向昇腾亲和算子编译时使用的中间表示,提供昇腾完备表达能力,通过编译优化提升昇腾AI处理器计算效率,支持通过生态框架使能昇腾AI处理器与深度调优
C++
503
346
cann-learning-hubcann-learning-hub
CANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。
Jupyter Notebook
787
415