安装stable-baselines3

问题描述

安装stable-baselines3一直显示不能正常运行
import stable_baselines3
一执行就报错
ModuleNotFoundError: No module named 'gymnasium…
TypeError: Too many arguments for <class ‘gymnasium.core.Wrapper’>; actual 4, expected 2

---

原因分析：

gym、gymnasium、stable-baselines3版本兼容问题

---

解决方案：

python 版本为3.10
gym 版本为0.17.0
gymnasium 版本为0.28.0
stable-baselines3 版本为2.3.2

---

小记：

安装pytorch可通过以下网址选出适合自己的安装语句

https://pytorch.org/

stable-baselines 只支持tensorflow 1.8.0-1.14.0，不支持2.x版本
tensorflow 1.x版本只支持3.7及以下版本的python

在python终端安装stable-baselines3

pip install stable-baselines3[extra]

相关官方网站

stable-baselines3:
https://github.com/DLR-RM/stable-baselines3?tab=readme-ov-file
https://araffin.github.io/post/sb3/
https://stable-baselines.readthedocs.io/en/master/guide/install.html#prerequisites\

gym官网讲解

gymnasium官网讲解