neon + gym + simple_DQN 简单搭建深度强化学习平台

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

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值