FROM mcr.microsoft.com/azureml/openmpi3.1.2-ubuntu18.04

USER root
RUN conda install -c anaconda python=3.7

RUN pip install ray-on-aml==0.1.6
RUN pip install gym[atari]==0.19.0
RUN pip install gym[accept-rom-license]==0.19.0
RUN pip install ale-py==0.7.0
RUN pip install azureml-core
RUN pip install ray==0.8.7
RUN pip install ray[rllib,tune,serve]==0.8.7
RUN pip install tensorflow==1.14.0

RUN apt-get update
RUN apt-get install -y jq
RUN apt-get install -y rsync