17 lines
342 B
Plaintext
17 lines
342 B
Plaintext
FROM python:3.6
|
|
|
|
RUN apt-get -y update && apt-get -y install ffmpeg libopenmpi-dev
|
|
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,mpi]
|
|
|
|
|
|
CMD /bin/bash
|