1 安装环境
- Ubuntu 15.04
- Anaconda(python 2.7)
- CUDA 7.5
- Opencv 3.0
2 安装
1 安装neon
Anaconda install
建立新的虚拟环境:
conda create --name neon pip
source activate neon
git neon,编译:
git clone https://github.com/NervanaSystems/neon.git /yourpath
cd neon && make sysinstall
2 安装OpenAI Gym
将完整gym安装作为python的一个库
pip install gym[all]
3 安装simple DQN
- 安装依赖
pip install numpy argparse logging
- 在虚拟环境中建立opencv软链接
sudo apt-get install python-opencv
ln -s /usr/lib/python2.7/dist-packages/cv2.so Anaconda_HOME/env/neon/lib/python2.7/site-packages/
- check out the code
git clone https://github.com/tambetm/simple_dqn /yourpath/
cd simple_dqn
4 Optional
For plotting install matplotlib:
pip install matplotlib
For producing game videos install avconv:
sudo apt-get install libav-tools
3 运行代码
1、训练
训练打砖块
./train.sh roms/breakout.bin