FROM nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu18.04 AS python-ubuntu

RUN apt-get update
RUN apt-get install -y build-essential curl python3 python3-pip python3-venv

# Inorder to run pip3.6, run it the following way `python3.6 -m pip`
# Update Python & Install wheel
RUN python3 -m pip install pip --upgrade && python3 -m pip install wheel

FROM python-ubuntu

RUN apt-get update &&  apt-get install -y  \
    liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git \
    pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
    libopenblas-dev libatlas-base-dev

RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow==2.4.4 && pip3 install tensorflow-gpu==2.4.4

WORKDIR /training

COPY scripts/train_incremental.py /usr/local/lib/python3.6/dist-packages/precise_lite/scripts/train_incremental.py
COPY entrypoint.sh /entrypoint.sh

ENTRYPOINT ["/entrypoint.sh"]
CMD ["/bin/bash"]