mirror of
https://github.com/wassname/retro-baselines.git
synced 2026-08-29 11:25:51 +08:00
18 lines
581 B
Docker
18 lines
581 B
Docker
FROM openai/retro-agent:tensorflow
|
|
|
|
# Needed for OpenCV.
|
|
RUN apt-get update && \
|
|
apt-get install -y libgtk2.0-dev && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Baselines has some unneeded and cumbersome dependencies,
|
|
# so we manually fetch the deps we need.
|
|
RUN . ~/venv/bin/activate && \
|
|
pip install scipy tqdm joblib zmq dill progressbar2 cloudpickle opencv-python && \
|
|
pip install --no-deps git+https://github.com/openai/baselines.git@24fe3d6576dd8f4cdd5f017805be689d6fa6be8c
|
|
|
|
ADD ppo2_agent.py ./agent.py
|
|
ADD sonic_util.py .
|
|
|
|
CMD ["python", "-u", "/root/compo/agent.py"]
|