lucasosouza / ptan

PyTorch Agent Net: reinforcement learning toolkit for pytorch

PTAN

PTAN stands for PyTorch AgentNet -- reimplementation of AgentNet library for PyTorch

This library was used in "Practical Deep Reinforcement Learning" book, here you can find sample sources.

Installation

From sources:

python setup.py install

From pypi:

pip install ptan

From github:

pip install pip install git+https://github.com/Shmuma/ptan.git 

Requirements

• PyTorch: version 0.4 is required, 0.3.1 is supported by old version ptan==0.2.1
• OpenAI Gym: pip install gym gym[atari]
• Python OpenCV: pip install opencv-python
• TensorBoard for PyTorch: pip install tensorboard-pytorch

Note for Anaconda Python users

To run some of the samples, you will need these modules:

conda install pytorch torchvision -c pytorch
pip install tensorboard-pytorch
pip install gym
pip install gym[atari]
pip install opencv-python