ab59de6922
* make baselines run without mpi wip * squash-merged latest master * further removing MPI references where unnecessary * more MPI removal * syntax and flake8 * MpiAdam becomes regular Adam if Mpi not present * autopep8 * add assertion to test in mpi_adam; fix trpo_mpi failure without MPI on cartpole * mpiless ddpg
17 lines
407 B
Docker
17 lines
407 B
Docker
FROM python:3.6
|
|
|
|
# RUN apt-get -y update && apt-get -y install git wget python-dev python3-dev libopenmpi-dev python-pip zlib1g-dev cmake python-opencv
|
|
ENV CODE_DIR /root/code
|
|
|
|
COPY . $CODE_DIR/baselines
|
|
WORKDIR $CODE_DIR/baselines
|
|
|
|
# Clean up pycache and pyc files
|
|
RUN rm -rf __pycache__ && \
|
|
find . -name "*.pyc" -delete && \
|
|
pip install tensorflow && \
|
|
pip install -e .[test]
|
|
|
|
|
|
CMD /bin/bash
|