Comments (2)
I haven't got a chance to fine tune it, but for now, I have it working. Based off official pytorch docker image, I just had to add the extra dependencies open AI gym mentions on their install page.
FROM pytorch/pytorch:v0.2
COPY . .
RUN apt-get update -qq && apt-get install -y --no-install-recommends \
python-numpy python-dev cmake zlib1g-dev libjpeg-dev xvfb libav-tools \
xorg-dev python-opengl libboost-all-dev libsdl2-dev swig \
&& apt-get clean && rm -rf /var/lib/apt/lists/*
RUN git clone https://github.com/openai/baselines.git && cd baselines && pip install -e .
RUN pip install -r requirements.txt
CMD ['/bin/sh']
So steps are:
- Clone this repo
- Download your trained models (see link in readme)
- Add the above into repo as file named
Dockerfile
- Build the image
docker build -t "pytorch-a2c-ppo-acktr" ./
- Run the docker image. Here we 'enjoy' pong:
nvidia-docker run --rm -it -v "$(pwd)/trained_models":/workspace/trained_models \
--user=$(id -u) \
--env="DISPLAY" \
--volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" \
awokeknowing/pytorch-a2c-ppo-acktr \
python enjoy.py --algo acktr --load-dir trained_models/acktr --env-name "PongNoFrameskip-v4" --num-stack 4
Again, you might consider having the built image at docker hub so that you could get your 'volunteers' up and running with just cloning the repo and running a script like bash volunteer.sh
(assuming they have nvidia-docker installed)
Also, note that it works with regular 'docker' with identical command. nvidia-docker just makes sure gpu is available, which you'd want for training.
I'll update as I do some more testing.
from pytorch-a2c-ppo-acktr-gail.
I added requirements to the readme file. Let me know if anything is missing.
from pytorch-a2c-ppo-acktr-gail.
Related Issues (20)
- does mask introduce bias in the gail implementation ?
- observation reset before insert
- Why acktr algorithm cannot be used in Mujoco settings?
- Can I train in my own game
- Can not run enjoy.py
- Stale hidden states
- Possible bug on the sign of policy log prob. in Fisher computation
- CNN Architecture
- Operations that have no effect
- No softmax before categorical loss?
- object has no attribute 'steps' in acktr
- why PPO needs to store action_log_probs instead of using stop_gradient for better efficiency? HOT 1
- [Question]Can I use Recurrent_policy for GAIL at this implementation?
- question about the recurrent HOT 1
- Oops! wrong repo :-D HOT 1
- setup.py and requirements.py have same dependencies except for h5py
- Where are the experts data for GAIL get from?
- Why is episode_rewards negative when running ant_v3 with PPO?
- Why didn't run to generate log?
- Updates: Support the latest Atari environment and state entropy maximization-based exploration.
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from pytorch-a2c-ppo-acktr-gail.