Stable-Baselines3安装

ccqq`

已于 2022-02-16 11:23:22 修改

阅读量9.2k

点赞数 4

分类专栏：强化学习平台文章标签： python 强化学习

于 2022-02-16 11:22:37 首次发布

本文链接：https://blog.csdn.net/cclbl/article/details/122959321

版权

强化学习平台专栏收录该内容

2 篇文章 4 订阅

订阅专栏

windows安装gym会有点问题。

安装stable-baselines3

pip install stable-baselines3[extra] -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

如果出现问题：

FileNotFoundError: Could not find module 'XXX\python\lib\site-packages\atari_py\ale_interface\ale_c.dll' (or one of its dependencies). Try using the full path with constructor syntax.

分三步：
1、先卸载atari-py。

pip uninstall atari-py
2、再重新安装这个。

pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py
3、重新下载gym

pip install gym，注意gym的版本对应stable-baselines3。

安装完成后，运行下以下代码

import gym
from stable_baselines3.common.vec_env.dummy_vec_env import DummyVecEnv
from stable_baselines3.common.evaluation import evaluate_policy
from stable_baselines3 import A2C,DQN,DDPG,TD3,PPO

env = gym.make("CartPole-v0")
model = DQN("MlpPolicy", env, verbose=1)