OpenAI 发布了一个名为 Gymnasium 的新库,它应该取代 Gym 库。
有许多库都实现了支持健身房环境的强化学习算法,但是 Gymnasium 的界面略有变化。有没有支持 Gymnasium 的算法库?我尝试了 CleanRL、KerasRL、RLib 和其他一些方法,但它们都不起作用,目前唯一的方法是手动实现它们,我对吗?
Stable Baselines3 还没有支持 Gymnasium 的版本,但是这个拉取请求 https://github.com/DLR-RM/stable-baselines3/pull/780解释了如何安装和使用它gymnasium
.
首先安装该 PR 的版本:
$ pip install git+https://github.com/carlosluis/stable-baselines3@fix_tests
假设你有gymnasium
已经安装,你可以运行:
# Important step to override `gym` as `gymnasium`.
import sys
import gymnasium
sys.modules["gym"] = gymnasium
# Sample code which works
from stable_baselines3 import PPO
env = gymnasium.make("CartPole-v1", render_mode="rgb_array")
model = PPO("MlpPolicy", env, verbose=1)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)