AI Max容器中创建Stable Baselines3环境

安装Stable Baselines3

pip install stable-baselines3

不能使用 pip install stable-baselines3[extra]命令安装，以为这个命令会帮助我们安装AutoRom，自动下载Atari环境。但是AIMax容器无法连接外网的。因此需要我们自己手动下载一些额外的依赖。

安装Gym Atari 环境

pip install ale-py
pip install pyglet
pip install autorom
pip install -U requests[socks]#以为等下下载Atari游戏包需要socks5代理

修改AutoRom.py的代码，改为类似这样。

resp = requests.get('http://go.to', 
                    proxies=dict(http='socks5://user:pass@host:port',
                                 https='socks5://user:pass@host:port'))

然后安装Atari环境

AutoRom

这个时候可以测试一下，环境是否安装成功。

import gym
env = gym.make('Breakout-v0')
obs = env.reset()
print(obs.shape)  # (210, 160, 3)

安装其他依赖

pip install tensorboard
apt install libgl1-mesa-glx#在Docker容器中使用opencv需要这个依赖
pip install opencv-python

疑难杂症

object has no attribute ‘randint‘

说明安装的gym和ale-py版本不对。当前我的版本为

• ale-py==0.7.4
• gym==0.21.0
• AutoROM==0.4.2

AttributeError: ‘NoneType’ object has no attribute ‘cvtColor’

说明没有安装刚刚的libgl1-mesa-glx依赖

apt install libgl1-mesa-glx#在Docker容器中使用opencv需要这个依赖