ARG processor
ARG region
ARG suffix

FROM 763104351884.dkr.ecr.$region.amazonaws.com/pytorch-training:1.5.0-$processor-py36-$suffix

RUN apt-get update && apt-get install -y --no-install-recommends \
        build-essential \
        jq \
        ffmpeg \
        rsync \
        libjpeg-dev \
        libxrender1 \
        python3-opengl \
        wget \
        xvfb && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip

RUN pip install --no-cache-dir \
    Cython==0.29.7 \
    tabulate \
    tensorboardX \
    gputil \
    gym==0.12.1 \
    lz4 \
    opencv-python-headless==4.1.0.25 \
    PyOpenGL==3.1.0 \
    pyyaml \
    redis==3.3.2 \
    ray==0.8.5 \
    ray[tune]==0.8.5 \
    ray[rllib]==0.8.5 \
    scipy \
    psutil \
    setproctitle \
    dm-tree \
    tensorflow-probability

# https://github.com/aws/sagemaker-rl-container/issues/39
RUN pip install pyglet==1.3.2

RUN conda install -y python=3.6.13

# https://click.palletsprojects.com/en/7.x/python3/
ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8

# Copy workaround script for incorrect hostname
COPY lib/changehostname.c /

COPY lib/start.sh /usr/local/bin/start.sh
RUN chmod +x /usr/local/bin/start.sh

# Starts framework
ENTRYPOINT ["bash", "-m", "start.sh"]