pip install elegantrl==0.3.3
欢迎访问链接: https://pepy.tech/project/elegantrl
关于Elegantrl:
「小雅:ElegantRL」
“小”:只需安装 PyTorch 和 Matplotlib 用于 网络训练 以及 画图协助调参,整个库只有4个 py 文件,并且tutorial版将一直维持在1000行代码以内,保持与最新版使用相同的变量名与代码结构,把学习成本降到最低。
“雅”:代码优雅 可读性高,耦合度低 可移植性高。我们整理出DRL算法的基本模板,让它们都继承自一个基类。只要照着模板编写新算法,就能自动支持多进程训练。
高性能:ElegantRL在支持多进程训练后,她的最新版最要支持多GPU训练了。RL与DL的不同导致它的分布式不能照搬DL的多GPU训练模式。所以无法直接使用PyTorch 或TensorFlow 这些深度学习框架自带的多GPU训练模块。
更多简介: https://zhuanlan.zhihu.com/p/358560363