1. 运行环境介绍
- NVIDIA GTX 1070
- Ubuntu 16.04 x64
- CUDA 8.0.61
- cuDNN 5.1
- Python 3.4
- TensorFlow 1.2.0
- Keras 2.2.4(Keras-2.2.4-py2.py3-none-any.whl)
- OpenCV 3.1.0 for Python3.4(opencv-3.1.0-np111py34_1.tar.bz2)
- gym(gym-0.12.0.tar.gz)
2. 准备
S1. 下载源码并解压,地址:https://github.com/keon/deep-q-learning
S2. 若希望从头开始体验training过程,可以删除源码中save
文件夹中的所有weights文件(或者修改成别的文件名也可以)。
3. 运行
S1. 打开源码中的ddqn.py
文件,取消第102行、第116行和第117行的注释。
S2. 新建终端,cd到源码的根目录下,运行如下命令即可:
python ddqn.py
运行过程如下两幅动图所示: